Update app.py

app.py

@@ -18,7 +18,6 @@ with st.container():
     st.markdown('<div class="scrollable-div">', unsafe_allow_html=True)
     #Langchain memory in session cache 
     if 'memory' not in st.session_state:
-        st.write("Memory is initilizing ...")
         st.session_state.memory = demo_chat.demo_miny_memory(model)
 
     #Check if chat history exists in this session
@@ -26,6 +25,7 @@ with st.container():
         st.session_state.chat_history = [ ] #Initialize chat history
 
     if 'model' not in st.session_state:
+        st.write("Model added in state.")
         st.session_state.model = model
 
     #renders chat history
@@ -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=st.session_state.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"):