from transformers import pipeline,Conversation 
chatbot = pipeline(model="facebook/blenderbot-400M-distill")
import gradio as gr

message_list= []
response_list = []

def yes_man(message,history):
  conversion = chatbot(message)

  return conversion[0]['generated_text']



def yes_man(message,history):
  conversion = chatbot(message)

  return conversion[0]['generated_text']

gr.ChatInterface(
    yes_man,
    chatbot=gr.Chatbot(height=300),
    textbox=gr.Textbox(placeholder="Ask me a yes or no question", container=False, scale=7),
    title="A vanilla chatbot",
    description="Ask Yes Man any question",
    theme="soft",
    examples=["Hello", "Am I cool?", "Are tomatoes vegetables?"],
    cache_examples=True,
    retry_btn=None,
    undo_btn="Delete Previous",
    clear_btn="Clear",
).launch(share=True)