Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -79,14 +79,14 @@ def create_conversational_chain(vector_store):
|
|
| 79 |
|
| 80 |
memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
|
| 81 |
|
| 82 |
-
|
| 83 |
retriever=vector_store.as_retriever(search_kwargs={"k": 2}),
|
| 84 |
-
memory=memory)
|
| 85 |
-
|
| 86 |
chain = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff",
|
| 87 |
retriever=vector_store.as_retriever(search_kwargs={"k": 2}),
|
| 88 |
memory = memory)
|
| 89 |
-
return chain
|
| 90 |
|
| 91 |
def main():
|
| 92 |
# Initialize session state
|
|
|
|
| 79 |
|
| 80 |
memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
|
| 81 |
|
| 82 |
+
chain = ConversationalRetrievalChain.from_llm(llm=llm, chain_type='stuff',
|
| 83 |
retriever=vector_store.as_retriever(search_kwargs={"k": 2}),
|
| 84 |
+
memory=memory)
|
| 85 |
+
"""
|
| 86 |
chain = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff",
|
| 87 |
retriever=vector_store.as_retriever(search_kwargs={"k": 2}),
|
| 88 |
memory = memory)
|
| 89 |
+
return chain """
|
| 90 |
|
| 91 |
def main():
|
| 92 |
# Initialize session state
|