wt002 commited on
Commit
29e83e0
·
verified ·
1 Parent(s): 3b7eced

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -6
agent.py CHANGED
@@ -413,15 +413,11 @@ docs = [
413
  # Now extract texts
414
  texts = [doc.page_content for doc in docs]
415
 
416
- # Generate embeddings
417
- embeddings = [embedding_model.encode(text) for text in texts]
418
-
419
-
420
  # Initialize the embedding model
421
- #embedding_model = BERTEmbeddings()
422
 
423
  # Generate embeddings for each document
424
- #embeddings = [embedding_model.encode(text) for text in texts]
425
 
426
  # Create the FAISS index
427
  vector_store = FAISS.from_documents(docs, embedding_model)
 
413
  # Now extract texts
414
  texts = [doc.page_content for doc in docs]
415
 
 
 
 
 
416
  # Initialize the embedding model
417
+ embedding_model = BERTEmbeddings()
418
 
419
  # Generate embeddings for each document
420
+ embeddings = [embedding_model.encode(text) for text in texts]
421
 
422
  # Create the FAISS index
423
  vector_store = FAISS.from_documents(docs, embedding_model)