Update app.py

app.py

@@ -2,18 +2,18 @@ import gradio as gr
 from transformers import pipeline
 import torch
 
-# Load the conversational pipeline with the model
+# Load the text generation pipeline with the model
 chatbot = pipeline(
-    "conversational", 
+    "text-generation",
     model="facebook/blenderbot-400M-distill",
     device=0 if torch.cuda.is_available() else -1
 )
 
 def chat(user_input):
     try:
-        # Process the conversation with the model
-        response = chatbot({"text": user_input})
-        # Return the generated response
+        # Generate response
+        response = chatbot(user_input, max_length=100, do_sample=True)
+        # Return the generated text
         return response[0]['generated_text']
     except Exception as e:
         return f"An error occurred: {str(e)}"
@@ -23,8 +23,8 @@ iface = gr.Interface(
     fn=chat,
     inputs=gr.Textbox(lines=2, placeholder="Type your message here..."),
     outputs="text",
-    title="Local Chatbot using Transformers",
-    description="A simple chat application running a local model with Hugging Face Transformers."
+    title="Chatbot using Blenderbot",
+    description="A simple chat application using the Blenderbot model."
 )
 
 if __name__ == "__main__":