wt002 commited on
Commit
b81a0d7
·
verified ·
1 Parent(s): eb1832e

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -192,8 +192,8 @@ for task in tasks:
192
  embedding_model = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2")
193
 
194
  # Create FAISS VectorStore from documents
195
- #vector_store = FAISS.from_documents(docs, embedding_model)
196
- vector_store = FAISS.load_local("faiss_index", embedding_model)
197
 
198
  # Save the FAISS index locally
199
  vector_store.save_local("faiss_index")
 
192
  embedding_model = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2")
193
 
194
  # Create FAISS VectorStore from documents
195
+ vector_store = FAISS.from_documents(docs, embedding_model)
196
+ #vector_store = FAISS.load_local("faiss_index", embedding_model)
197
 
198
  # Save the FAISS index locally
199
  vector_store.save_local("faiss_index")