rohit commited on
Commit
8946f02
·
1 Parent(s): b5e33d3

remove logs

Browse files
Files changed (1) hide show
  1. app/pipeline.py +9 -7
app/pipeline.py CHANGED
@@ -53,19 +53,21 @@ class RAGPipeline:
53
  )
54
  self.documents.append(document)
55
 
56
- # print 10 documents
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