Hugging Face's logo

Spaces:
itachi-ai
/
Chat-Bot
Running

App Files Community
itachi-ai commited on
Commit
a3cd6cc
·
verified ·
1 Parent(s): 221b1c2

updated

Browse files
Files changed (1) hide show
  1. embed_with_db.py +1 -1
embed_with_db.py CHANGED
@@ -62,7 +62,7 @@ class VECTORDB_STORE:
62
  return None
63
 
64
  def chain(self):
65
- retriever = self.vectordb_store.as_retriever(search_kwargs={"k": 5})
66
  chain = {'context': retriever, 'question': RunnablePassthrough()} | prompt | llm_model | parser
67
  return chain
68
 
 
62
  return None
63
 
64
  def chain(self):
65
+ retriever = self.vectordb_store.as_retriever(search_kwargs={"k": 10})
66
  chain = {'context': retriever, 'question': RunnablePassthrough()} | prompt | llm_model | parser
67
  return chain
68