app.py

app.py

@@ -74,8 +74,8 @@ emb_filename = 'unsplash-25k-photos-embeddings.pkl'
 with open(emb_filename, 'rb') as fIn:
         img_names, img_emb = pickle.load(fIn)
  
-def display_matches(indices):
-    best_matched_images = [Image.open(IMAGES_DIR / img_names[top_k_best_image]) for top_k_best_image in torch.topk(similarity, 2, 0).indices]
+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]
     return best_matched_images
   
 def image_search(search_text, search_image, option):