app.py

app.py

@@ -71,11 +71,15 @@ model = SentenceTransformer('clip-ViT-B-32')
 
 # pre-computed embeddings
 emb_filename = 'unsplash-25k-photos-embeddings.pkl'
-with open(emb_filename, 'rb') as fIn:
-        img_names, img_emb = pickle.load(fIn)
+with open(emb_filename, 'rb') as emb:
+        img_names, img_emb = pickle.load(emb)
  
 def display_matches(similarity):
-    best_matched_images = [Image.open(IMAGES_DIR / img_names[top_k_best_image]) for top_k_best_image in torch.topk(similarity, 4, 0).indices]
+    best_matched_images = []
+    top_k_indices = torch.topk(similarity, 4, 0).indices
+    for matched_image in top_k_indices:
+        img = Image.open(IMAGES_DIR / img_names[matched_image])
+        best_matched_images.append(img)
     return best_matched_images
   
 def image_search(search_text, search_image, option):