Update demo/modules/search.py

demo/modules/search.py

@@ -74,13 +74,12 @@ def search(input: str, nprobe: int, topk: int, input_type: str, query_type: str,
         index = all_index[query_type][db]["index"]
         ids = all_index[query_type][db]["ids"]
         
-    if check_index_ivf(query_type, subsection_type):
+    if hasattr(index, "nprobe"):
         if index.nlist < nprobe:
             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}).")
         else:
             index.nprobe = nprobe
 
-    print(nprobe)
     if topk > index.ntotal:
         raise gr.Error(f"You cannot retrieve more than the database size ({index.ntotal}).")