Update app.py

app.py

@@ -427,7 +427,9 @@ def get_beam_search_html(
     length_penalty=-3.0,
     num_return_sequences=3
 ):
-    inputs = tokenizer([input_text], return_tensors="pt")
+    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")
 
     outputs = model.generate(
         **inputs,