Update app.py

app.py

@@ -22,28 +22,35 @@ with gr.Blocks(theme=gr.themes.Ocean(primary_hue="pink", neutral_hue="indigo", f
         button = gr.Button("Describir", variant="primary")
         clear = gr.Button("Borrar")
     output = gr.Textbox(label="Resumen")
+    with gr.Row():
+        button2 = gr.Button("Leer", variant="primary")
+        clear = gr.Button("Borrar")
+    output2 = gr.Audio(label="Audio")
 
     def describir(url):
       raw_image = Image.open(requests.get(url, stream=True).raw).convert('RGB')
       inputs = processor(raw_image, return_tensors="pt").to("cpu")
       out = model.generate(**inputs)
-      models, cfg, task = load_model_ensemble_and_task_from_hf_hub(
-      "facebook/fastspeech2-en-ljspeech",
-      arg_overrides={"vocoder": "hifigan", "fp16": False}
-      )
-      ble = models[0]
-      TTSHubInterface.update_cfg_with_data_cfg(cfg, task.data_cfg)
-      generator = task.build_generator(ble, cfg)
-        
-      text = processor.decode(out[0], skip_special_tokens=True)
+      return processor.decode(out[0], skip_special_tokens=True)
+
+    def leer(texto):
+        models, cfg, task = load_model_ensemble_and_task_from_hf_hub(
+        "facebook/fastspeech2-en-ljspeech",
+        arg_overrides={"vocoder": "hifigan", "fp16": False}
+        )
+        model = models[0]
+        TTSHubInterface.update_cfg_with_data_cfg(cfg, task.data_cfg)
+        generator = task.build_generator(model, cfg)
         
-      sample = TTSHubInterface.get_model_input(task, text)
-      wav, rate = TTSHubInterface.get_prediction(task, ble, generator, sample)
-      ipd.Audio(wav, rate=rate)
+        text = texto
         
-      return processor.decode(out[0], skip_special_tokens=True)
+        sample = TTSHubInterface.get_model_input(task, text)
+        wav, rate = TTSHubInterface.get_prediction(task, model, generator, sample)
+        ipd.Audio(wav, rate=rate)
+        return None
 
 
     button.click(describir, [textbox], output)
+    button2.click(leer, [output], output2)
 
 demo.launch(debug=True)