Update app.py

app.py

@@ -427,7 +427,7 @@ def get_beam_search_html(
     length_penalty=-3.0,
     num_return_sequences=3
 ):
-    input_with_template=f"<|im_start|>{input_text}"
+    input_with_template=f"<|im_start|>system /n You are a helpful chatbot.<|im_end|><|im_start|>{input_text}"
 
     inputs = tokenizer([input_with_template], return_tensors="pt")
 
@@ -491,7 +491,7 @@ with gr.Blocks(
     n_beams=1
     length_penalty=1
     num_return_sequences=3
-    button = gr.Button()
+    button = gr.Button("Decode")
     out_html = gr.Markdown()
     out_markdown = gr.Markdown()
     button.click(