RobotJelly commited on
Commit
4d9725d
·
1 Parent(s): e59258f
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -94,9 +94,9 @@ def image_search(search_text, search_image, option):
94
  image_feature /= image_feature.norm(dim=-1, keepdim=True)
95
  image_feature = image_feature.cpu().numpy()
96
  # Find the matched Images
97
- matched_images = find_matches(image_feature, photo_features, photo_ids, 4)
98
  #is_input_image = True
99
- return show_output_image(matched_images)
100
 
101
  gr.Interface(fn=image_search,
102
  inputs=[gr.inputs.Textbox(lines=7, label="Input Text"),
 
94
  image_feature /= image_feature.norm(dim=-1, keepdim=True)
95
  image_feature = image_feature.cpu().numpy()
96
  # Find the matched Images
97
+ matched_images = find_matches(image_feature, photo_features, photo_ids, 4)
98
  #is_input_image = True
99
+ return show_output_image(matched_images)
100
 
101
  gr.Interface(fn=image_search,
102
  inputs=[gr.inputs.Textbox(lines=7, label="Input Text"),