Spaces:

sivan22
/

Semantic-Search-upload-your-file

Runtime error

sivan22 commited on May 28, 2024

Commit

422aa66

1 Parent(s): 3d33fb5

use CPU

Files changed (1) hide show

app.py CHANGED Viewed

@@ -31,7 +31,7 @@ def get_embeddings(df,_embeddings_model) ->object:
 @st.cache_resource
 def get_model()->object:
     model_name = "intfloat/multilingual-e5-large"
-    model_kwargs = {'device': 'cuda'} #'cpu' or 'cuda'
     encode_kwargs = {'normalize_embeddings': True}
     embeddings_model = HuggingFaceEmbeddings(
         model_name=model_name,

 @st.cache_resource
 def get_model()->object:
     model_name = "intfloat/multilingual-e5-large"
+    model_kwargs = {'device': 'cpu'} #'cpu' or 'cuda'
     encode_kwargs = {'normalize_embeddings': True}
     embeddings_model = HuggingFaceEmbeddings(
         model_name=model_name,