Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -43,8 +43,8 @@ class DocumentRetrievalAndGeneration:
|
|
| 43 |
all_texts = [split.page_content for split in self.all_splits]
|
| 44 |
embeddings = self.embeddings.encode(all_texts, convert_to_tensor=True).cpu().numpy()
|
| 45 |
vectordb = FAISS.from_embeddings(
|
| 46 |
-
embeddings,
|
| 47 |
-
self.embeddings,
|
| 48 |
metadatas=[{"source": f"doc_{i}"} for i in range(len(all_texts))]
|
| 49 |
)
|
| 50 |
return vectordb
|
|
|
|
| 43 |
all_texts = [split.page_content for split in self.all_splits]
|
| 44 |
embeddings = self.embeddings.encode(all_texts, convert_to_tensor=True).cpu().numpy()
|
| 45 |
vectordb = FAISS.from_embeddings(
|
| 46 |
+
text_embeddings=list(zip(all_texts, embeddings)), # Pairing texts with embeddings
|
| 47 |
+
model=self.embeddings,
|
| 48 |
metadatas=[{"source": f"doc_{i}"} for i in range(len(all_texts))]
|
| 49 |
)
|
| 50 |
return vectordb
|