Update app.py

app.py

@@ -3,7 +3,9 @@ import gradio as gr
 
 # Load the pipeline for text generation
 try:
-    generator = pipeline("text-generation", model="microsoft/DialoGPT-small")
+ #   generator = pipeline("text-generation", model="microsoft/DialoGPT-small")
+    generator = pipeline("text-generation", model="microsoft/DialoGPT-medium")
+
 except Exception as e:
     print(f"Error loading the model: {e}")
     raise