Update app.py

app.py

@@ -14,10 +14,6 @@ st.write("I am your hotel booking assistant for today.")
 
 # tokenizer = AutoTokenizer.from_pretrained("KvrParaskevi/Hotel-Assistant-Attempt4-Llama-2-7b")
 
-# [theme]
-# base="light"
-# primaryColor="#6b4bff"
-
 tokenizer, model = demo_chat.load_model()
 
 model_identifier = "KvrParaskevi/Hotel-Assistant-Attempt4-Llama-2-7b"
@@ -60,7 +56,7 @@ with st.container():
 
     if input_text:
         with st.chat_message("user"):
-            st.write(input_text)
+            #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= chat_model)
@@ -71,6 +67,6 @@ with st.container():
         first_answer = tokenizer.decode(outputs[0][tokenized_chat.shape[1]:],skip_special_tokens=True)
         
         with st.chat_message("assistant"):
-            st.write(first_answer)
+            #st.write(first_answer)
             st.session_state.chat_history.append({"role": "assistant", "content": first_answer})
     st.markdown('</div>', unsafe_allow_html=True)