Spaces:
Running
Running
Update src/doc_qa.py
Browse files- 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 |
|