# -*- 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() | |