Aditya757864 commited on
Commit
87079be
·
verified ·
1 Parent(s): cf566f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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
- """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
 
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