import gradio as gr
from transformers import pipeline

# Load the model (example using GPT-2)
generator = pipeline('text-generation', model='gpt2')

def generate_text(prompt):
    result = generator(prompt, max_length=100, num_return_sequences=1)
    return result[0]['generated_text']

# Create the Gradio interface
iface = gr.Interface(
    fn=generate_text,
    inputs="text",
    outputs="text",
    title="Text Generation API"
)

# Launch the app
iface.launch()