app.py

app.py

@@ -87,7 +87,7 @@ def image_search(search_text, search_image, option):
     #text_features = encode_search_query(search_text, model, device)
      text_emb = model.encode([search_text], convert_to_tensor=True)
      similarity = util.cos_sim(img_emb, text_emb)
-     return [Image.open(img_folder / img_names[top_k_best_image]) for top_k_best_image in torch.topk(similarity, 2, 0).indices]
+     return [Image.open(IMAGES_DIR / img_names[top_k_best_image]) for top_k_best_image in torch.topk(similarity, 2, 0).indices]
     # Find the matched Images
     #matched_images = find_matches(text_features, photo_features, photo_ids, 4)
      #matched_results = util.semantic_search(text_emb, img_emb, top_k=4)[0]