chayanbhansali commited on
Commit
ffb2c37
·
verified ·
1 Parent(s): e050927

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -51,7 +51,7 @@ class RAGChatbot:
51
  self.documents.extend(chunks)
52
 
53
  # Generate embeddings
54
- self.embeddings = self.embedding_model.encode(self.documents, batch_size=32, show_progress_bar=True)
55
  return f"Loaded {len(self.documents)} text chunks from {len(file_paths)} files"
56
 
57
  def retrieve_relevant_context(self, query, top_k=3):
 
51
  self.documents.extend(chunks)
52
 
53
  # Generate embeddings
54
+ self.embeddings = self.embedding_model.encode(self.documents)
55
  return f"Loaded {len(self.documents)} text chunks from {len(file_paths)} files"
56
 
57
  def retrieve_relevant_context(self, query, top_k=3):