Update app.py

app.py

@@ -143,8 +143,8 @@ from transformers import AutoTokenizer,AutoModelForCausalLM
 
 template = """
 {history}
-: {human_input}
-:"""
+<human>: {human_input}
+<bot>:"""
 
 prompt = PromptTemplate(
     input_variables=["history", "human_input"], 
@@ -259,10 +259,10 @@ def list2prompt(history):
 
 PROMPT_DICT = {
     "prompt_input": (
-        ": {input}\n: {instruction}\n: "
+        "<context>: {input}\n<human>: {instruction}\n<bot>: "
     ),
     "prompt_no_input": (
-        ": {instruction}\n: "
+        "<human>: {instruction}\n<bot>: "
     ),
 }