Spaces:
Sleeping
Sleeping
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 =
|
| 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 |
)
|