Update app.py
Browse files
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
|
| 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 |
|