app.py

app.py

@@ -34,7 +34,7 @@ def display_matches(similarity, topk):
     return best_matched_images
   
 def image_search(Option, topk, search_text, search_image):
-  
+  topk = topk+1
   # Input Text Query
   if Option == "Text-To-Image" :
     # Encode the given Input  text for Search & take it in tensor form
@@ -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+1)
+     return display_matches(similarity, topk)
   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+1)
+      return display_matches(similarity, topk)
   
 gr.Interface(fn=image_search, title="Search Image",
              description="Enter the text or image to search for the most relevant images...",