Arun-AK commited on
Commit
6adde50
·
verified ·
1 Parent(s): c98b165

Update pages/2_🧠_context_aware_chatbot.py

Browse files
pages/2_🧠_context_aware_chatbot.py CHANGED
@@ -32,6 +32,7 @@ class ContextChatbot:
32
  './Docs/sickle-cell-disease-report.txt',
33
  './Docs/Blood Basics.txt',
34
  './Docs/Sickle Cell Basic Info.txt'
 
35
  ]
36
 
37
  # Load documents
@@ -77,7 +78,7 @@ class ContextChatbot:
77
  llm = llm,
78
  chain_type = "stuff",
79
  memory = ConversationSummaryMemory(llm = summarizer, memory_key='chat_history', input_key='question', output_key= 'answer', return_messages=True),
80
- retriever = vectorstore.as_retriever(k = 3, search_type="mmr"),
81
  return_source_documents=False,
82
  combine_docs_chain_kwargs=chain_type_kwargs
83
  )
 
32
  './Docs/sickle-cell-disease-report.txt',
33
  './Docs/Blood Basics.txt',
34
  './Docs/Sickle Cell Basic Info.txt'
35
+ './Docs/Manging Sicklecell.txt'
36
  ]
37
 
38
  # Load documents
 
78
  llm = llm,
79
  chain_type = "stuff",
80
  memory = ConversationSummaryMemory(llm = summarizer, memory_key='chat_history', input_key='question', output_key= 'answer', return_messages=True),
81
+ retriever = vectorstore.as_retriever(k = 5, search_type="mmr"),
82
  return_source_documents=False,
83
  combine_docs_chain_kwargs=chain_type_kwargs
84
  )