Update app.py

app.py

@@ -51,6 +51,8 @@ model_name = 'hackathon-pln-es/poem-gen-spanish-t5-small'
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 
+tts_es = gr.Interface.load("huggingface/facebook/tts_transformer-es-css10")
+
 def make_poem(author, sentiment, words, text):
   num_lines=5
   poem = text
@@ -83,7 +85,8 @@ def make_poem(author, sentiment, words, text):
 
     pre_output = pre_output.replace('.', ' ')
     poem += '\n' + pre_output
-  return poem
+  audio = tts_es(poem)
+  return poem, audio
 
 iface = gr.Interface(
     fn=make_poem, 
@@ -94,7 +97,7 @@ iface = gr.Interface(
      gr.inputs.Textbox(lines=1, placeholder='palabra_1, palabra_2, ..., palabra_n', label='Palabras que desea incluir'),
      gr.inputs.Textbox(lines=4, placeholder='texto inicial', label='Texto inicial'),
      ],
-     outputs="text",
+     outputs=["text", "audio"],
      examples=
      [
       ['Pablo Neruda', 'positivo', 'cielo, luna, mar', 'Todos fueron a verle pasar']