Nelly-43 commited on
Commit
7f738ef
·
verified ·
1 Parent(s): e4bbf63

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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'