from transformers import pipeline
import gradio as gr

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

def generate_text(prompt):
    return generator(prompt, max_length=50)[0]["generated_text"]

# Interface with title and description
interface = gr.Interface(
    fn=generate_text,
    inputs=gr.Textbox(label="Your prompt here"),
    outputs=gr.Textbox(label="Generated response"),
    title="Mini Text Generator",
    description="Type the start of a sentence, and I'll complete it for you! (Note: Works best in English, French results might be poor.)"
)

interface.launch()