app.py

app.py

@@ -79,7 +79,7 @@ with open(emb_filename, 'rb') as fIn:
  
 def display_matches(similarity):
     best_matched_images = []
-    for best_img in torch.topk(similarity, 4, 0).indices:
+    for best_img in torch.topk(similarity, 4).indices:
          img = Image.open(os.path.join('./photos', img_names[best_img]))
          best_matched_images.append(img)
     return best_matched_images