app.py

app.py

@@ -75,7 +75,7 @@ with open(emb_filename, 'rb') as fIn:
         img_names, img_emb = pickle.load(fIn)
  
 def display_matches(indices):
-    best_matched_images = [Image.open(os.path.join(IMAGES_DIR, img_names[best_img['corpus_id']])) for best_img in indices]
+    best_matched_images = [Image.open(os.path.join("photos/", img_names[best_img['corpus_id']])) for best_img in indices]
     return best_matched_images
   
 def image_search(search_text, search_image, option):
@@ -106,8 +106,8 @@ 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.fromarray(search_image)], convert_to_tensor=True)
-  
+      #image_emb = model.encode(Image.open(search_image), convert_to_tensor=True)
+      image_emb = model.encode(Image.open(search_image))
       # Find the matched Images
       #matched_images = find_matches(text_features, photo_features, photo_ids, 4)
       #similarity = util.cos_sim(image_emb, img_emb)