Spaces:
Runtime error
Runtime error
Update agent.py
Browse files
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 |
-
|
422 |
|
423 |
# Generate embeddings for each document
|
424 |
-
|
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)
|