Update app.py

app.py

@@ -2,6 +2,9 @@ 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 generate_response(user_input):
     prompt = formatted_prompt(user_input)