Update app.py

app.py

@@ -1,11 +1,15 @@
+import gradio as gr
 from transformers import pipeline
 
-# Use a more specific conversational model, like DialoGPT
+# Use the text generation pipeline with DialoGPT-medium
 chatbot = pipeline("text-generation", model="microsoft/DialoGPT-medium")
 
 def chat(input_text):
     response = chatbot(input_text, max_length=50, num_return_sequences=1)
     return response[0]['generated_text']
 
-# Test the function with an example input
-print(chat("How are you feeling today?"))
+# Gradio Interface setup
+interface = gr.Interface(fn=chat, inputs="text", outputs="text")
+
+# Launch the interface
+interface.launch()