Update app.py

app.py

@@ -11,10 +11,7 @@ st.write("I am your hotel booking assistant for today.")
 # base="light"
 # primaryColor="#6b4bff"
 
-model_Loading = True
-if model_Loading == True:
-    model = demo_chat.get_Model()
-    model_Loading = False
+model = demo_chat.get_Model()
 
 #Application 
 with st.container():
@@ -28,6 +25,9 @@ with st.container():
     if 'chat_history' not in st.session_state:
         st.session_state.chat_history = [ ] #Initialize chat history
 
+    if 'model' not in st.session_state:
+        st.session_state.model = model
+
     #renders chat history
     for message in st.session_state.chat_history: 
         with st.chat_message(message["role"]):
@@ -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= 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"):