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