Update app.py
Browse files
app.py
CHANGED
|
@@ -44,7 +44,7 @@ chat_llm = ChatHuggingFace(llm=model, verbose=True)
|
|
| 44 |
model_name = "sentence-transformers/all-mpnet-base-v2"
|
| 45 |
model_kwargs = {"device": "cuda"}
|
| 46 |
|
| 47 |
-
embeddings = HuggingFaceEmbeddings(model_name=model_name, model_kwargs=model_kwargs)
|
| 48 |
|
| 49 |
def docs_return(directory_path, flag):
|
| 50 |
docx_file_pattern = '*.docx'
|
|
|
|
| 44 |
model_name = "sentence-transformers/all-mpnet-base-v2"
|
| 45 |
model_kwargs = {"device": "cuda"}
|
| 46 |
|
| 47 |
+
embeddings = HuggingFaceEmbeddings(model_name=model_name)#, model_kwargs=model_kwargs)
|
| 48 |
|
| 49 |
def docs_return(directory_path, flag):
|
| 50 |
docx_file_pattern = '*.docx'
|