LTEnjoy commited on
Commit
63c4ae7
·
verified ·
1 Parent(s): 43977ca

Update demo/modules/search.py

Browse files
Files changed (1) hide show
  1. 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