from transformers import pipeline, Conversation import gradio as gr chatbot = pipeline(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'] demo_chatbot = gr.ChatInterface(vanilla_chatbot, title="Mashdemy Chatbot", description="Enter text to start chatting.") demo_chatbot.launch(share = True)