islasher commited on
Commit
270c181
1 Parent(s): ffd1749

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -31,8 +31,8 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
31
  all_splits = text_splitter.split_documents(documents)
32
 
33
  model_name = "sentence-transformers/paraphrase-multilingual-mpnet-base-v2"
34
- model_kwargs = {"device": "cuda"}
35
- embeddings = HuggingFaceEmbeddings(model_name=model_name, model_kwargs=model_kwargs)
36
 
37
  vectordb = Chroma.from_documents(documents=all_splits, embedding=embeddings, persist_directory="chroma_db")
38
 
 
31
  all_splits = text_splitter.split_documents(documents)
32
 
33
  model_name = "sentence-transformers/paraphrase-multilingual-mpnet-base-v2"
34
+ # model_kwargs = {"device": "cuda"}
35
+ embeddings = HuggingFaceEmbeddings(model_name=model_name)#, model_kwargs=model_kwargs)
36
 
37
  vectordb = Chroma.from_documents(documents=all_splits, embedding=embeddings, persist_directory="chroma_db")
38