app.py

app.py

@@ -90,7 +90,7 @@ def image_search(search_text, search_image, option):
   
     # Find the matched Images
     #matched_images = find_matches(text_features, photo_features, photo_ids, 4)
-     matched_results = util.semantic_search(text_emb, img_emb, 4)[0]
+     matched_results = util.semantic_search(text_emb, img_emb, top_k=4)[0]
     
     # top 4 highest ranked images
      return display_matches(matched_results)
@@ -106,7 +106,7 @@ def image_search(search_text, search_image, option):
       # Find the matched Images
       #matched_images = find_matches(image_feature, photo_ids, 4)
       
-      image_emb = model.encode(Image.open(search_image), convert_to_tensor=True)
+      image_emb = model.encode([Image.fromarray(search_image)], convert_to_tensor=True)
   
       # Find the matched Images
       #matched_images = find_matches(text_features, photo_features, photo_ids, 4)