singhdevendra58 commited on
Commit
e8418ff
·
verified ·
1 Parent(s): a554343

Update src/doc_qa.py

Browse files
Files changed (1) hide show
  1. src/doc_qa.py +2 -4
src/doc_qa.py CHANGED
@@ -26,10 +26,8 @@ multi_directory_path=r'tmp/'
26
  from transformers import pipeline
27
 
28
  from sentence_transformers import SentenceTransformer
29
- model = SentenceTransformer("sentence-transformers/LaBSE")
30
-
31
- embeddings = HuggingFaceEmbeddings(model_name='sentence-transformers/LaBSE')
32
-
33
 
34
  from langchain_community.document_loaders import TextLoader, PyPDFLoader, Docx2txtLoader
35
 
 
26
  from transformers import pipeline
27
 
28
  from sentence_transformers import SentenceTransformer
29
+ #model = SentenceTransformer("sentence-transformers/LaBSE")
30
+ embeddings = HuggingFaceEmbeddings(model_name='setu4993/LaBSE')
 
 
31
 
32
  from langchain_community.document_loaders import TextLoader, PyPDFLoader, Docx2txtLoader
33