Spaces:
Runtime error
Runtime error
| import gradio as gr | |
| from transformers import pipeline | |
| generator = pipeline("text-generation", model="gpt2") | |
| def generate_text(prompt): | |
| text = generator(prompt, max_length=50)[0]['generated_text'] | |
| return text | |
| iface = gr.Interface(fn=generate_text, | |
| inputs=gr.inputs.Textbox(lines=5, label="Enter prompt here"), | |
| outputs=gr.outputs.Textbox(label="Generated text")) | |
| iface.launch() | |