Update app.py

app.py

@@ -2,7 +2,7 @@ from transformers import pipeline
 import gradio as gr
 
 # Initialize the text generation pipeline
-generator = pipeline("text-generation", model="EleutherAI/gpt-neo-1.3B")
+generator = pipeline("text-generation", model="distilgpt2")
 
 def generate_response(prompt):
     """Generate text based on the user's prompt."""