Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import pipeline
 
-generator = pipeline("text-generation", model="microsoft/DialoGPT-small")
+generator = pipeline("text-generation", model="distilgpt2")
 
 def chat_fn(message, history=None):
     try:
@@ -10,4 +10,5 @@ def chat_fn(message, history=None):
         return bot_reply, history or []
     except Exception as e:
         return f"Error: {e}", history or []
+
 gr.ChatInterface(fn=chat_fn).launch()