Update app.py

app.py

@@ -1,10 +1,10 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 
-sonnets_pretrained_model = "datificate/gpt2-small-spanish"
-sonnets_tokenizer = AutoTokenizer.from_pretrained(sonnets_pretrained_model, use_fast=True)
+pretrained_model = "datificate/gpt2-small-spanish"
+tokenizer = AutoTokenizer.from_pretrained(pretrained_model, use_fast=True)
 sonnets_tuned_model = 'hackathon-pln-es/gpt2-small-spanish-disco-poetry'
-sonnets_pipe = pipeline('text2text-generation', model=sonnets_tuned_model, tokenizer=sonnets_tokenizer)
+sonnets_pipe = pipeline('text2text-generation', model=sonnets_tuned_model, tokenizer=tokenizer)
 
 def make_new_sonnet(prompt, max_lenght):
     ouputs = sonnets_pipe(prompt, max_length=max_lenght, 
@@ -14,17 +14,34 @@ def make_new_sonnet(prompt, max_lenght):
                   num_return_sequences=1)
     return ouputs[0]['generated_text']
 
-article = "<p style='text-align: center'>Don't miss this other cool space based in a model of different styles of poetry in spanish: <a href='https://huggingface.co/spaces/hackathon-pln-es/poem-generation-es' target='_blank'>poem-generation-es</a></p>"
+
+description = """
+<p>Este experimento sólo pretende poner a prueba la capacidad de un modelo de inteligencia artificial para encontrar patrones en la hechicería del lenguaje poético en español.
+Algo de lo que Borges habla en una conferencia de su Arte Poética:
+<br/>
+<i>“Uso la palabra «modelo» porque las metáforas que voy a citar, aunque parezcan muy distintas a la imaginación, para un lógico serían casi idénticas. Así que podríamos hablar de ellas como ecuaciones.”</i>
+</p>
+"""
+article = """
+<p>
+<h3>¿Por qué sonetos?</h3>
+Este experimento se basa en un modelo gpt-2 entrenado con un corpus exclusivo de sonetos del siglo 15 al 20 en español.
+La la elección de los sonetos obedece a una razón fundamental: su estilo y métrica específica, buscando entre otras razones, tener una consistencia en la experimentación.
+</p>
+<p style='text-align: center'> 👉 Don't miss this other cool space based on different styles of poetry in spanish: <a href='https://huggingface.co/spaces/hackathon-pln-es/poem-generation-es' target='_blank'>poem-generation-es</a> 👈 </p>
+"""
+
 examples = [
-    ['vendrá la muerte y tendrá tus ojos',140],
-    ['buenas cosas pasan cuando anochece', 200],
-    ['al despertar el dinosaurio todavía estaba ahí', 140 ]
+    ['al despertar el dinosaurio todavía estaba ahí', 140 ],
+    ['La inmensidad de tu imaginación', 140],
+    ['con ansias de amores inflamada', 140]
 ]
 iface = gr.Interface(fn=make_new_sonnet,
-                      title= "Generador de poesía basada en sonetos en español",
+                      title= "Creador de poesía basada en sonetos en español",
+                      description = description,
                       inputs=[
                               gr.inputs.Textbox(lines=2, placeholder="Escrbe algo para comenzar", label='Escribe algo para comenzar'),
-                              gr.inputs.Slider(minimum = 30, maximum = 200, default = 140, step = 10, label='Salida de caracteres')],
+                              gr.inputs.Slider(minimum = 60, maximum = 200, default = 140, step = 10, label='Salida máxima de caracteres')],
                       outputs=[
                                gr.outputs.Textbox(label="Tu poema"),
                               ],