import gradio as gr
from transformers import pipeline

# Load a very small text-generation model
generator = pipeline("text-generation", model="distilgpt2")

def generate(text):
    result = generator(text, max_length=60, num_return_sequences=1)
    return result[0]["generated_text"]

gr.Interface(
    fn=generate,
    inputs=gr.Textbox(label="Write a sentence"),
    outputs=gr.Textbox(label="AI output"),
    title="Tiny Text Generator"
).launch()