Update app.py

app.py

@@ -1,8 +1,8 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load DistilBERT for text generation on CPU
-model = pipeline("text-generation", model="distilbert-base-uncased", device="cpu")
+# Load distilgpt2 for text generation on CPU
+model = pipeline("text-generation", model="distilgpt2", device="cpu")
 
 # Define text generation function
 def generate_text(message, history):
@@ -15,9 +15,9 @@ def generate_text(message, history):
 # Gradio ChatInterface
 interface = gr.ChatInterface(
     fn=generate_text,
-    title="DistilBERT Chatbot",
-    description="Powered by DistilBERT on Hugging Face Spaces (CPU Basic). Use API or UI!",
-    examples=["Hello, how are you?", "Tell me a short fact", "What’s 2+2?"]
+    title="DistilGPT2 Chatbot",
+    description="Powered by DistilGPT2 on Hugging Face Spaces (CPU Basic). Use API or UI!",
+    examples=["Hello, how are you?", "Tell me a short joke", "What’s 2+2?"]
 )
 
 # Launch with API endpoint