Update app.py

app.py

@@ -1,8 +1,7 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load a small text-generation model
-generator = pipeline("text-generation", model="gpt2")
+generator = pipeline("text-generation", model="distilgpt2")
 
 def chat(prompt):
     result = generator(prompt, max_length=100, num_return_sequences=1)