rohit
commited on
Commit
·
8946f02
1
Parent(s):
b5e33d3
remove logs
Browse files- app/pipeline.py +9 -7
app/pipeline.py
CHANGED
|
@@ -53,19 +53,21 @@ class RAGPipeline:
|
|
| 53 |
)
|
| 54 |
self.documents.append(document)
|
| 55 |
|
| 56 |
-
|
| 57 |
-
for doc in self.documents[:10]:
|
| 58 |
-
print(f"Content: {doc.content}")
|
| 59 |
-
print(f"Metadata: {doc.meta}")
|
| 60 |
-
print("-"*100)
|
| 61 |
|
| 62 |
# Initialize components
|
| 63 |
self.document_store = InMemoryDocumentStore()
|
| 64 |
self.doc_embedder = SentenceTransformersDocumentEmbedder(
|
| 65 |
-
model=embedding_model or MODEL_CONFIG["embedding_model"]
|
|
|
|
| 66 |
)
|
| 67 |
self.text_embedder = SentenceTransformersTextEmbedder(
|
| 68 |
-
model=embedding_model or MODEL_CONFIG["embedding_model"]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 69 |
)
|
| 70 |
self.retriever = InMemoryEmbeddingRetriever(self.document_store)
|
| 71 |
|
|
|
|
| 53 |
)
|
| 54 |
self.documents.append(document)
|
| 55 |
|
| 56 |
+
# Documents loaded silently - remove verbose output
|
|
|
|
|
|
|
|
|
|
|
|
|
| 57 |
|
| 58 |
# Initialize components
|
| 59 |
self.document_store = InMemoryDocumentStore()
|
| 60 |
self.doc_embedder = SentenceTransformersDocumentEmbedder(
|
| 61 |
+
model=embedding_model or MODEL_CONFIG["embedding_model"],
|
| 62 |
+
progress_bar=False
|
| 63 |
)
|
| 64 |
self.text_embedder = SentenceTransformersTextEmbedder(
|
| 65 |
+
model=embedding_model or MODEL_CONFIG["embedding_model"],
|
| 66 |
+
progress_bar=False
|
| 67 |
+
)
|
| 68 |
+
self.text_embedder = SentenceTransformersTextEmbedder(
|
| 69 |
+
model=embedding_model or MODEL_CONFIG["embedding_model"],
|
| 70 |
+
progress_bar=False
|
| 71 |
)
|
| 72 |
self.retriever = InMemoryEmbeddingRetriever(self.document_store)
|
| 73 |
|