from transformers import pipeline
import gradio as gr

generator = pipeline(
    "text-generation",
    model="distilgpt2"
)

def generate(text):
    result = generator(text, max_new_tokens=50)
    return result[0]["generated_text"]

gr.Interface(
    fn=generate,
    inputs=gr.Textbox(lines=5),
    outputs="text",
    title="distilgpt2 Demo"
).launch()