Update app.py

app.py

@@ -46,7 +46,8 @@ if input_text := st.chat_input(placeholder="Here you can chat with our hotel boo
     with st.spinner("Generating response..."):
         #first_answer = llm_chain.predict(input = input_text)
         #answer = first_answer.strip()
-        input = "Human:" + input_text + "Assistant:"
+        prompt = demo_chat.chat_template_prompt()
+        input = prompt + input_text + "Assistant:"
         answer = re.generate_response(input)
         
         with st.chat_message("assistant"):