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'
 #def show_output_image(matched_images) :
    #image=[]
    #for photo_id in matched_images:
@@ -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("photos/", img_names[best_img['corpus_id']])) for best_img in indices]
+    best_matched_images = [Image.open(os.path.join(IMAGES_DIR, img_names[best_img['corpus_id']])) for best_img in indices]
     return best_matched_images
   
 def image_search(search_text, search_image, option):