Spaces:
Sleeping
Sleeping
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)
|