seayala commited on
Commit
eb277be
·
verified ·
1 Parent(s): b48bd08

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=20
26
  all_splits = text_splitter.split_documents(documents)
27
 
28
  model_name = "sentence-transformers/paraphrase-multilingual-mpnet-base-v2"
29
- model_kwargs = {"device": "cuda"}
30
 
31
  embeddings = HuggingFaceEmbeddings(model_name=model_name, model_kwargs=model_kwargs)
32
 
 
26
  all_splits = text_splitter.split_documents(documents)
27
 
28
  model_name = "sentence-transformers/paraphrase-multilingual-mpnet-base-v2"
29
+ model_kwargs = {"device": "cpu"}
30
 
31
  embeddings = HuggingFaceEmbeddings(model_name=model_name, model_kwargs=model_kwargs)
32