Spaces:
Running
Running
Update vlog4chat.py
Browse files- vlog4chat.py +3 -1
vlog4chat.py
CHANGED
@@ -25,7 +25,8 @@ from langchain.llms.base import LLM
|
|
25 |
from langchain.chains import ConversationalRetrievalChain
|
26 |
from langchain_core.output_parsers import StrOutputParser
|
27 |
from langchain.text_splitter import RecursiveCharacterTextSplitter
|
28 |
-
from langchain.llms import HuggingFacePipeline
|
|
|
29 |
#from langchain.embeddings import HuggingFaceEmbeddings
|
30 |
from langchain_community.embeddings import HuggingFaceEmbeddings
|
31 |
from langchain_community.document_loaders import TextLoader
|
@@ -146,6 +147,7 @@ class Vlogger4chat :
|
|
146 |
self.history = []
|
147 |
if not os.path.exists(self.tmp_dir):
|
148 |
os.makedirs(self.tmp_dir)
|
|
|
149 |
|
150 |
def init_llm(self):
|
151 |
print('\033[1;33m' + "Initializing LLM Reasoner...".center(50, '-') + '\033[0m')
|
|
|
25 |
from langchain.chains import ConversationalRetrievalChain
|
26 |
from langchain_core.output_parsers import StrOutputParser
|
27 |
from langchain.text_splitter import RecursiveCharacterTextSplitter
|
28 |
+
#from langchain.llms import HuggingFacePipeline
|
29 |
+
from langchain_community.llms import HuggingFacePipeline
|
30 |
#from langchain.embeddings import HuggingFaceEmbeddings
|
31 |
from langchain_community.embeddings import HuggingFaceEmbeddings
|
32 |
from langchain_community.document_loaders import TextLoader
|
|
|
147 |
self.history = []
|
148 |
if not os.path.exists(self.tmp_dir):
|
149 |
os.makedirs(self.tmp_dir)
|
150 |
+
self.my_embedding = HuggingFaceEmbeddings(model_name='BAAI/bge-m3', model_kwargs={'device': 'cpu'} ,encode_kwargs={'normalize_embeddings': True})
|
151 |
|
152 |
def init_llm(self):
|
153 |
print('\033[1;33m' + "Initializing LLM Reasoner...".center(50, '-') + '\033[0m')
|