Spaces:
Build error
Build error
RobotJelly
commited on
Commit
·
4d9725d
1
Parent(s):
e59258f
app.py
Browse files
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 |
-
|
98 |
#is_input_image = True
|
99 |
-
|
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"),
|