Spaces:

GIZ
/

GIZ-Project-Search

Running on CPU Upgrade

annikwag commited on Jan 7

Commit

5cd34a4

verified ·

1 Parent(s): eac610a

Update appStore/search.py

Files changed (1) hide show

appStore/search.py CHANGED Viewed

@@ -6,7 +6,7 @@ from langchain_huggingface import HuggingFaceEmbeddings
 device = 'cuda' if cuda.is_available() else 'cpu'
-def hybrid_search(client, query, collection_name, limit=3000):
     embeddings = HuggingFaceEmbeddings(
         model_name='BAAI/bge-m3',
         model_kwargs={'device': device},

 device = 'cuda' if cuda.is_available() else 'cpu'
+def hybrid_search(client, query, collection_name, limit=500):
     embeddings = HuggingFaceEmbeddings(
         model_name='BAAI/bge-m3',
         model_kwargs={'device': device},