Update app.py

app.py

@@ -14,10 +14,10 @@ from datasets import load_dataset
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 def load_models_and_data():
-    model_name = "microsoft/speecht5_tts"
+    model_name = "Moustapha91/TTS_WOLOF_FINAL"
     processor = SpeechT5Processor.from_pretrained(model_name)
-    model = SpeechT5ForTextToSpeech.from_pretrained("Moustapha91/speecht5_tts__v3_sn").to(device)
-    vocoder = SpeechT5HifiGan.from_pretrained("microsoft/speecht5_hifigan").to(device)
+    model = SpeechT5ForTextToSpeech.from_pretrained("model_name").to(device)
+    vocoder = SpeechT5HifiGan.from_pretrained("model_name").to(device)
 
     spk_model_name = "speechbrain/spkrec-xvect-voxceleb"
     speaker_model = EncoderClassifier.from_hparams(