| import gradio as gr | |
| import transformers | |
| from transformers import pipeline | |
| generator = pipeline('text-generation', model='huggingtweets/arthur_rimbaud') | |
| def generate_text(input_text): | |
| generated_text = generator(input_text, num_return_sequences=1)[0]['generated_text'] | |
| return generated_text | |
| iface = gr.Interface(fn=generate_text, inputs="text", outputs="text", title="Rim-bot vuelve a casa") | |
| iface.launch() |