Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import streamlit as st
 import model as demo_chat
+import request as re
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
 st.title("Hi, I am Chatbot Philio :woman:")
@@ -18,7 +19,7 @@ scrollable_div_style = """
 </style>
 """
 
-llm_chain = demo_chat.chain()
+#llm_chain = demo_chat.chain()
 
 def render_chat_history(chat_history):
     #renders chat history
@@ -43,8 +44,10 @@ if input_text := st.chat_input(placeholder="Here you can chat with our hotel boo
     st.session_state.chat_history.append({"role" : "human", "content" : input_text}) #append message to chat history
 
     with st.spinner("Generating response..."):
-        first_answer = llm_chain.predict(input = input_text)
-        answer = first_answer.strip()
+        #first_answer = llm_chain.predict(input = input_text)
+        #answer = first_answer.strip()
+        input = input_text + "AI:"
+        answer = re.generate_response(input)
         
         with st.chat_message("assistant"):
             st.markdown(answer)