Spaces:
Running
Running
Update vlog4chat.py
Browse files- vlog4chat.py +2 -2
vlog4chat.py
CHANGED
@@ -158,8 +158,8 @@ class Vlogger4chat :
|
|
158 |
#self.vector_storage = FAISS.from_documents(chunks, self.my_embedding)
|
159 |
pkl_path = os.path.join(self.data_dir, f"{video_id}.pkl")
|
160 |
with open(pkl_path, 'rb') as f:
|
161 |
-
vector_storage =pickle.load(f)
|
162 |
-
|
163 |
self.chain = ConversationalRetrievalChain.from_llm(self.llm, self.vector_storage.as_retriever(), return_source_documents=True)
|
164 |
return True
|
165 |
return False
|
|
|
158 |
#self.vector_storage = FAISS.from_documents(chunks, self.my_embedding)
|
159 |
pkl_path = os.path.join(self.data_dir, f"{video_id}.pkl")
|
160 |
with open(pkl_path, 'rb') as f:
|
161 |
+
self.vector_storage = pickle.load(f)
|
162 |
+
|
163 |
self.chain = ConversationalRetrievalChain.from_llm(self.llm, self.vector_storage.as_retriever(), return_source_documents=True)
|
164 |
return True
|
165 |
return False
|