speech-to-speech-translation

Running

GatinhoEducado commited on 7 days ago

Commit

13f8a3e

•

1 Parent(s): 0bb8a04

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,7 +26,7 @@ processor = SpeechT5Processor.from_pretrained("GatinhoEducado/speechT5_tts-finet
 model = SpeechT5ForTextToSpeech.from_pretrained("GatinhoEducado/speechT5_tts-finetuned-cml-tts").to(device)
 vocoder = SpeechT5HifiGan.from_pretrained("microsoft/speecht5_hifigan").to(device)
-embeddings_dataset = load_dataset("Matthijs/cmu-arctic-xvectors", split="validation")
 speaker_embeddings = torch.tensor(embeddings_dataset[7306]["xvector"]).unsqueeze(0)

 model = SpeechT5ForTextToSpeech.from_pretrained("GatinhoEducado/speechT5_tts-finetuned-cml-tts").to(device)
 vocoder = SpeechT5HifiGan.from_pretrained("microsoft/speecht5_hifigan").to(device)
+embeddings_dataset = load_dataset("Matthijs/cmu-arctic-xvectors", split="validation", trust_remote_code=True)
 speaker_embeddings = torch.tensor(embeddings_dataset[7306]["xvector"]).unsqueeze(0)