Update demo/modules/search.py

demo/modules/search.py

@@ -100,7 +100,7 @@ def search(input: str, nprobe: int, topk: int, input_type: str, query_type: str,
         raise gr.Error(f"You cannot retrieve more than the database size ({index.ntotal}).")
     
     # Retrieve all scores to plot the distribution
-    scores, ranks = index.search(input_embedding, max(topk, nprobe*256))
+    scores, ranks = index.search(input_embedding, index.ntotal)
     scores, ranks = scores[0], ranks[0]
     
     # Remove inf values