Update app.py

app.py

@@ -1,14 +1,17 @@
-from transformers import pipeline, Conversation
+from transformers import pipeline
 import gradio as gr
 
-chatbot = pipeline(model="facebook/blenderbot-400M-distill")
+chatbot = pipeline("conversational", model="facebook/blenderbot-400M-distill")
+
 message_list = []
 response_list = []
 
 def vanilla_chatbot(message, history):
- conversation = Conversation(text=message, past_user_inputs=message_list,generated_responses=response_list)
- bot = chatbot(conversation.messages[0]['content']) # working code
- return bot[-1]['generated_text']
-    
+    global message_list, response_list
+    message_list.append(message)
+    bot_response = chatbot(message_list)
+    response_list.append(bot_response[-1]['generated_text'])
+    return bot_response[-1]['generated_text']
+
 demo_chatbot = gr.ChatInterface(vanilla_chatbot, title="Mashdemy Chatbot", description="Enter text to start chatting.")
-demo_chatbot.launch(share = True)
+demo_chatbot.launch(share=True)