Update app.py

app.py

@@ -1,31 +1,27 @@
-from transformers import Conversation, pipeline
+from transformers import pipeline
 import gradio as gr
 
 # Load the chatbot pipeline
-chatbot = pipeline(model="facebook/blenderbot-400M-distill")
+chatbot = pipeline("conversational", model="facebook/blenderbot-400M-distill")
 
-# Initialize message history lists
-message_list = []
-response_list = []
+# Initialize message history
+conversation_history = []
 
 # Function to interact with the chatbot
 def vanilla_chatbot(message, history):
-    # Create a Conversation object
-    conversation = Conversation(
-        text=message,
-        past_user_inputs=message_list,
-        generated_responses=response_list
-    )
+    global conversation_history
+    
+    # Create a conversation turn
+    conversation_history.append({"role": "user", "content": message})
     
     # Generate bot response
-    bot_response = chatbot(conversation.messages[0]['content'])
+    bot_response = chatbot(conversation_history)
     
-    # Append message and response to history lists
-    message_list.append(message)
-    response_list.append(bot_response[0]['generated_text'])
+    # Append bot response to history
+    conversation_history.append({"role": "bot", "content": bot_response.generated_responses[-1]})
     
     # Return the generated response
-    return bot_response[0]['generated_text']
+    return bot_response.generated_responses[-1]
 
 # Create a Gradio chat interface
 demo_chatbot = gr.Interface(