Update app.py

app.py

@@ -1,22 +1,20 @@
 import gradio as gr
-from transformers import pipeline, Conversation
+from transformers import pipeline
 import torch
 
 # Load the conversational pipeline with the model
 chatbot = pipeline(
     "conversational", 
     model="facebook/blenderbot-400M-distill",
-    device=0 if torch.cuda.is_available() else -1  # Use GPU if available
+    device=0 if torch.cuda.is_available() else -1
 )
 
 def chat(user_input):
     try:
-        # Initialize a conversation with the user's message
-        conversation = Conversation(user_input)
         # Process the conversation with the model
-        updated_conversation = chatbot(conversation)
-        # Return the latest generated response
-        return updated_conversation.generated_responses[-1]
+        response = chatbot({"text": user_input})
+        # Return the generated response
+        return response[0]['generated_text']
     except Exception as e:
         return f"An error occurred: {str(e)}"