Leo Trieu
Initialize app
b0324cd
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()