Update app.py

app.py

@@ -1,7 +1,7 @@
 import torch
 from transformers import pipeline
 from datasets import load_dataset
-from transformers import SpeechT5Processor, SpeechT5ForConditionalGeneration, SpeechT5HifiGan
+from transformers import SpeechT5Processor, Speech2Text2Speech, SpeechT5HifiGan
 import numpy as np
 import gradio as gr
 
@@ -17,7 +17,7 @@ def translate(audio):
 
 # Cargar el procesador y el modelo de SpeechT5
 processor = SpeechT5Processor.from_pretrained("microsoft/speecht5_tts")
-model = SpeechT5ForConditionalGeneration.from_pretrained("gitgato/mabama")
+model = Speech2Text2Speech.from_pretrained("gitgato/mabama")
 vocoder = SpeechT5HifiGan.from_pretrained("microsoft/speecht5_hifigan")
 
 # Cargar los datos de embeddings del hablante