Update app.py

app.py

@@ -13,11 +13,11 @@ from transformers import SpeechT5Processor, SpeechT5ForTextToSpeech, SpeechT5Hif
 
 processor = SpeechT5Processor.from_pretrained("microsoft/speecht5_tts")
 
-model = SpeechT5ForTextToSpeech.from_pretrained("microsoft/speecht5_tts")
+model = SpeechT5ForTextToSpeech.from_pretrained("gitgato/mabama")
 vocoder = SpeechT5HifiGan.from_pretrained("microsoft/speecht5_hifigan")
 
-embeddings_dataset = load_dataset("Matthijs/cmu-arctic-xvectors", split="validation")
-speaker_embeddings = torch.tensor(embeddings_dataset[144]["gitgato/mabama"]).unsqueeze(0)
+embeddings_dataset = load_dataset("ovieyra21/mabama-v5", split="validation")
+speaker_embeddings = torch.tensor(embeddings_dataset[144]["xvector"]).unsqueeze(0)
 
 def synthesise(text):
     inputs = processor(text=text, return_tensors="pt")