Update app.py

app.py

@@ -1,8 +1,11 @@
 import gradio as gr
+from transformers import pipeline
 
-def greet(name):
-    return "Hello " + name + "!"
+generator = pipeline('text-generation', model='huggingtweets/arthur_rimbaud')
 
-demo = gr.Interface(fn=greet, inputs="textbox", outputs="textbox")
-    
-demo.launch(share=True)  # Share your demo with just 1 extra parameter 🚀
+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()