Spaces:
Runtime error
Runtime error
Arno' Francesco (GDS DS&G) commited on
Commit ·
53820cc
1
Parent(s): d6b8788
fix docs
Browse files- retriever.py +5 -3
retriever.py
CHANGED
|
@@ -34,11 +34,13 @@ class GuestInfoRetrieverTool(Tool):
|
|
| 34 |
|
| 35 |
#self.model = model
|
| 36 |
|
| 37 |
-
docs_for_embeddings = ['passage: '+d.page_content for d in docs]
|
| 38 |
-
|
|
|
|
|
|
|
| 39 |
#embeddings = self.model.encode(docs_for_embeddings, normalize_embeddings=True)
|
| 40 |
|
| 41 |
-
self.vector_store = FAISS.from_documents(
|
| 42 |
self.faiss_retriever = self.vector_store.as_retriever(search_kwargs={"k": 5})
|
| 43 |
|
| 44 |
|
|
|
|
| 34 |
|
| 35 |
#self.model = model
|
| 36 |
|
| 37 |
+
#docs_for_embeddings = ['passage: '+d.page_content for d in docs]
|
| 38 |
+
|
| 39 |
+
for d in docs:
|
| 40 |
+
d.page_content = 'passage: '+d.page_content
|
| 41 |
#embeddings = self.model.encode(docs_for_embeddings, normalize_embeddings=True)
|
| 42 |
|
| 43 |
+
self.vector_store = FAISS.from_documents(docs, hf)
|
| 44 |
self.faiss_retriever = self.vector_store.as_retriever(search_kwargs={"k": 5})
|
| 45 |
|
| 46 |
|