# -*- coding: utf-8 -*-

from transformers import pipeline, set_seed

generator = pipeline('text-generation', model='gpt2')

import gradio as gr
def func(text):
  generated  = generator(text)
  return generated

app = gr.Interface(fn=func, inputs="textbox", outputs="textbox", title="Text Generator")
app.launch()