Spaces:
Build error
Build error
Commit
·
aa88f3a
1
Parent(s):
927e71f
app.py
Browse files
app.py
CHANGED
@@ -112,7 +112,7 @@ def image_search(search_text, search_image, option):
|
|
112 |
#matched_images = find_matches(text_features, photo_features, photo_ids, 4)
|
113 |
#similarity = util.cos_sim(image_emb, img_emb)
|
114 |
#matched_results = util.semantic_search(image_emb, img_emb, 4)[0]
|
115 |
-
image_emb = model.encode([Image.
|
116 |
similarity = util.cos_sim(img_emb, image_emb)
|
117 |
return display_matches(similarity)
|
118 |
|
|
|
112 |
#matched_images = find_matches(text_features, photo_features, photo_ids, 4)
|
113 |
#similarity = util.cos_sim(image_emb, img_emb)
|
114 |
#matched_results = util.semantic_search(image_emb, img_emb, 4)[0]
|
115 |
+
image_emb = model.encode([Image.open(search_image)], convert_to_tensor=True)
|
116 |
similarity = util.cos_sim(img_emb, image_emb)
|
117 |
return display_matches(similarity)
|
118 |
|