Update app.py

app.py

@@ -25,7 +25,7 @@ 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.session_state.memory = demo_chat.demo_miny_memory(model)
+        st.session_state.memory = demo_chat.demo_miny_memory(chat_model)
 
     #Check if chat history exists in this session
     if 'chat_history' not in st.session_state:
@@ -39,6 +39,8 @@ with st.container():
     for message in st.session_state.chat_history: 
         with st.chat_message(message["role"]):
             st.write(message["content"])
+            
+    chat_model._to_chat_prompt(st.session_state.chat_history)
 
     #Set up input text field
     input_text = st.chat_input(placeholder="Here you can chat with our hotel booking model.")