Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from transformers import AutoModelForCausalLM, AutoTokenizer
+from transformers import AutoModelForCausalLLM, AutoTokenizer
 from langchain.llms.base import LLM
 from langchain import PromptTemplate, LLMChain
 
@@ -23,9 +23,23 @@ class CustomLLM(LLM):
 
 llm = CustomLLM()
 
-template = """Question: {question}
-Answer: Let's think step by step."""
-prompt = PromptTemplate(template=template, input_variables=["question"])
+template = """<<SYS>>
+You are an AI having conversation with a human. Below is an instruction that describes a task. 
+Write a response that appropriately completes the request.
+Reply with the most helpful and logic answer. During the conversation you need to ask the user 
+the following questions to complete the hotel booking task.
+1) Where would you like to stay and when?
+2) How many people are staying in the room?
+3) Do you prefer any ammenities like breakfast included or gym?
+4) What is your name, your email address and phone number? 
+Make sure you receive a logical answer from the user from every question to complete the hotel 
+booking process.
+<</SYS>>
+Previous conversation:
+{history}
+Human: {input}
+AI:"""
+prompt = PromptTemplate(template=template, input_variables=["history", "input"])
 
 llm_chain = LLMChain(prompt=prompt, llm=llm)