Update chatbot.py

chatbot.py

@@ -26,7 +26,7 @@ booking process.
 Current conversation:
 {history}
 
-Human: {{{input}}}
+Human: {input}
 AI: [/INST]"""
 
 #@st.cache_resource 
@@ -68,7 +68,7 @@ def demo_miny_memory():
 
 def demo_chain(input_text,history):
     #PROMPT = ChatPromptTemplate.from_template(template)
-    PROMPT = ChatPromptTemplate.from_messages(history)
+    PROMPT = PromptTemplate(template=template, input_variables=["history", "input"])
     conversation = ConversationChain(
         prompt=PROMPT,
         llm=llm,
@@ -78,4 +78,4 @@ def demo_chain(input_text,history):
     )
 
     chat_reply = conversation.invoke(input=input_text, max_new_tokens = 100)
-    return chat_reply['response']
+    return chat_reply.split('AI:')[-1]