app.py

import gradio as gr
from transformers import pipeline

text_generator = pipeline(task="text-generation")

def generate_text(prompt):
  output = text_generator(
    prompt, 
    max_length=100, 
    num_return_sequences=1
  )
  return output[0]["generated_text"]

iface = gr.Interface(
    fn=generate_text,
    inputs="textbox",
    outputs="label",
    title="Text Generation",
)

# Launch the Gradio app
iface.launch()