app.py

app.py

@@ -113,7 +113,7 @@ def image_search(search_text, search_image, option):
       #similarity = util.cos_sim(image_emb, img_emb)
       #matched_results = util.semantic_search(image_emb, img_emb, 4)[0]
       
-      image_emb = model.encode([Image.open(search_image)], convert_to_tensor=True)
+      image_emb = model.encode([Image.fromarray(search_image)], convert_to_tensor=True)
       similarity = util.cos_sim(img_emb, image_emb)
       return display_matches(similarity)