Spaces:
Build error
Build error
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"),
|