app.py

app.py

@@ -43,7 +43,7 @@ def image_search(Option, topk, search_text, search_image):
      similarity = util.cos_sim(img_emb, text_emb)
      
      #using the computed similarities, find the topk best matches
-     return display_matches(similarity, topk)
+     return display_matches(similarity, topk+1)
   elif Option == "Image-To-Image":
       # Encode the given Input Image for Search & take it in tensor form
       image_emb = model.encode([Image.fromarray(search_image)], convert_to_tensor=True)
@@ -51,7 +51,7 @@ def image_search(Option, topk, search_text, search_image):
       similarity = util.cos_sim(img_emb, image_emb)
       
       #using the computed similarities, find the topk best matches
-      return display_matches(similarity, topk)
+      return display_matches(similarity, topk+1)
   
 gr.Interface(fn=image_search, title="Search Image",
              description="Enter the text or image to search the other most relevant images...",