from transformers import pipeline
from transformers import Conversation

import gradio as gr

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

import gradio as gr

message_list = []
response_list = []

def mini_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(
    mini_chatbot,
    title="My Chatbot",
    description="Enter text to start chatting."
)

demo_chatbot.launch()