Update app.py

app.py

@@ -1,14 +1,17 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load a conversational model from Hugging Face Hub
+# Load the chatbot model using text-generation pipeline
 chatbot = pipeline("text-generation", model="microsoft/DialoGPT-medium")
 
 def respond(message, chat_history):
+    # Generate a response using the model
+    response = chatbot(message, max_new_tokens=100)[0]['generated_text']
+    # Update chat history
     chat_history = chat_history or []
     chat_history.append(("User", message))
-    response = chatbot(message)[0]['generated_text']
     chat_history.append(("Bot", response))
+    # Format chat history for display
     chat_display = ""
     for speaker, text in chat_history:
         chat_display += f"{speaker}: {text}\n"
@@ -20,7 +23,11 @@ with gr.Blocks() as demo:
     user_input = gr.Textbox(label="Your Message")
     send_button = gr.Button("Send")
 
-    send_button.click(fn=respond, inputs=[user_input, chat_history], outputs=[chatbot_output, chat_history])
+    send_button.click(
+        fn=respond,
+        inputs=[user_input, chat_history],
+        outputs=[chatbot_output, chat_history]
+    )
 
 if __name__ == "__main__":
     demo.launch()