app.py

app.py

@@ -31,7 +31,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 # taking features vectors
 #photo_features = np.load("./features.npy")
 
-#IMAGES_DIR = './photos'
+IMAGES_DIR = './photos/'
 
 
 
@@ -79,8 +79,9 @@ with open(emb_filename, 'rb') as fIn:
  
 def display_matches(similarity):
     best_matched_images = []
-    for best_img in torch.topk(similarity, 4).indices:
-         img = Image.open(os.path.join('./photos', img_names[best_img]))
+    indices = torch.topk(input=similarity, k=4, dim=0)
+    for best_img_idx in indices:
+         img = Image.open(IMAGES_DIR / img_names[best_img_idx])
          best_matched_images.append(img)
     return best_matched_images