Update app.py

app.py

@@ -41,7 +41,7 @@ with st.container():
             st.write(input_text)
             st.session_state.chat_history.append({"role" : "user", "content" : input_text}) #append message to chat history
 
-        chat_response = demo_chat.demo_chain(input_text=input_text, memory=st.session_state.memory, model=st.session_state.model)
+        chat_response = demo_chat.demo_chain(input_text=input_text, memory=st.session_state.memory, model= model)
         first_answer = chat_response.split("Human")[0] #Because of Predict it prints the whole conversation.Here we seperate the first answer only.
 
         with st.chat_message("assistant"):