Update app.py

app.py

@@ -5,7 +5,7 @@ import torch
 
 tokenizer = AutoTokenizer.from_pretrained("Hieu-Pham/Llama2-7B-QLoRA-cooking-text-gen-merged")
 
-stop_token_ids = tokenizer.convert_tokens_to_ids(["\n", "#", "\\", "`", "###", "##", "Question", "Comment", "Answer"])
+stop_token_ids = tokenizer.convert_tokens_to_ids(["\n", "#", "\\", "`", "###", "##", "Question", "Comment", "Answer", "Context"])
 
 # define custom stopping criteria object
 class StopOnTokens(StoppingCriteria):
@@ -43,7 +43,7 @@ demo = gr.Interface(
     predict,
     inputs=[gr.Textbox(lines=2, placeholder="Please provide your question", label="Question"), gr.Textbox(lines=2, placeholder="Please provide your context", label="Context")],
     outputs=gr.Textbox(lines=2, placeholder="Answers...", label="Output"),
-    title="Question Answering",
+    title="Cooking Recipe MRC",
 )
 
 demo.launch()