Update app.py

app.py

@@ -2,8 +2,8 @@ from time import perf_counter
 import streamlit as st
 from transformers import pipeline, AutoModelForSequenceClassification, AutoTokenizer, GenerationConfig
 
-def formatted_train(input,output)->str:
-  return f"<|im_start|>user\n{input}|im_end|>\n|im_start|>assistant\n{output}<|im_end|>\n"
+def formatted_prompt(input)-> str:
+  return f"<|im_start|>user\n{input}<|im_end|>\n<|im_start|>assistant:"
 
 def generate_response(user_input):
     prompt = formatted_prompt(user_input)