import gradio as gr
from transformers import pipeline

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

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

demo = gr.Interface(
    fn=generate_text,
    inputs=gr.Textbox(placeholder="Enter prompt here"),
    outputs=gr.Textbox(),
    title="AI Text Generator"
)

demo.launch()