from transformers import pipeline
import gradio as gr

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


def vanilla_chatbot(message, history):
    conversation = chatbot(message)
    
    return conversation[0]['generated_text']

demo_chatbot = gr.ChatInterface(vanilla_chatbot, title="POWO Chatbot", description="Entre un texte pour commencer le chatting.")

demo_chatbot.launch()