Update app.py

app.py

@@ -1,5 +1,13 @@
 import gradio as gr
 from gradio import mix
+from transformers import pipeline
+
+generator = pipeline("text-generation", model="distilgpt2")
+generator(
+    "In this course, we will teach you how to",
+    max_length=30,
+    num_return_sequences=2,
+)
 
 
 title = "Miniature"
@@ -16,7 +24,7 @@ io = gr.Interface.load("huggingface/keras-io/text-generation")
 
 def inference(text):
 
-    return io(text)
+    return generator(text)