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 |
|