import gradio as gr

from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration

tokenizer = BlenderbotTokenizer.from_pretrained('facebook/blenderbot-1B-distill')
model = BlenderbotForConditionalGeneration.from_pretrained('facebook/blenderbot-1B-distill')

def func (message):
  inputs = tokenizer(message, return_tensors='pt')
  result = model.generate(**inputs)
  return tokenizer.decode(result[0])
  
app = gr.Interface(fn=func, inputs = 'textbox', outputs = 'textbox', title='Chatbot Three')
app.launch()