Update app.py

app.py

@@ -1,9 +1,6 @@
 import gradio as gr
 from transformers import pipeline, AutoTokenizer
 
-def load_model(model_name):
-    return pipeline("text-generation", model=model_name, device="cpu")
-
 def generate(
     model_name,
     system_input,
@@ -11,7 +8,7 @@ def generate(
     assistant_initial_message,
     user_input,
 ):
-    pipe = load_model(model_name)
+    pipe = pipeline("text-generation", model=model_name, device="cpu")
     
     message_template = [
         {"role": "system", "content": system_input},