Update demo/modules/search.py
Browse files- demo/modules/search.py +2 -1
demo/modules/search.py
CHANGED
@@ -79,7 +79,8 @@ def search(input: str, nprobe: int, topk: int, input_type: str, query_type: str,
|
|
79 |
raise gr.Error(f"The number of clusters to search must be less than or equal to the number of clusters in the index ({index.nlist}).")
|
80 |
else:
|
81 |
index.nprobe = nprobe
|
82 |
-
|
|
|
83 |
if topk > index.ntotal:
|
84 |
raise gr.Error(f"You cannot retrieve more than the database size ({index.ntotal}).")
|
85 |
|
|
|
79 |
raise gr.Error(f"The number of clusters to search must be less than or equal to the number of clusters in the index ({index.nlist}).")
|
80 |
else:
|
81 |
index.nprobe = nprobe
|
82 |
+
|
83 |
+
print(nprobe)
|
84 |
if topk > index.ntotal:
|
85 |
raise gr.Error(f"You cannot retrieve more than the database size ({index.ntotal}).")
|
86 |
|