from transformers import pipeline
from transformers import Conversation
import gradio as gr

message_list = []
response_list = []

chatbot = pipeline("conversational", model = "facebook/blenderbot-400M-distill")

def vanilla_chatbot(message, history):
    conversation = Conversation(text = message, past_user_inputs = message_list, generated_responses = response_list)
    conversation = chatbot(conversation)
    return conversation.generated_responses[-1]

demo_chatbot = gr.ChatInterface(vanilla_chatbot, title = "Vanilla Chatbot", description = "Enter text to start chatting.")
demo_chatbot.launch()