Spaces:
Sleeping
Sleeping
File size: 368 Bytes
b0324cd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
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() |