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 Generator" ) iface.launch()