Update app.py

app.py

@@ -1,14 +1,14 @@
 import torch
 from transformers import pipeline
 from datasets import load_dataset
+from transformers import AutoModel
 from transformers import SpeechT5Processor, Speech2Text2Speech, SpeechT5HifiGan
 import numpy as np
 import gradio as gr
 
 # Configurar el pipeline de reconocimiento automático de voz
-pipe = pipeline(
-    "automatic-speech-recognition", model="openai/whisper-base"
-)
+pipe = pipeline("automatic-speech-recognition", model="openai/whisper-base")
+# Load model directly
 
 # Función para traducir texto
 def translate(audio):
@@ -17,7 +17,7 @@ def translate(audio):
 
 # Cargar el procesador y el modelo de SpeechT5
 processor = SpeechT5Processor.from_pretrained("microsoft/speecht5_tts")
-model = Speech2Text2Speech.from_pretrained("gitgato/mabama")
+model = AutoModel.from_pretrained("gitgato/mabama")
 vocoder = SpeechT5HifiGan.from_pretrained("microsoft/speecht5_hifigan")
 
 # Cargar los datos de embeddings del hablante