speechbrain
/

tts-hifigan-unit-hubert-l6-k100-ljspeech

speech-synthesis

Model card Files Files and versions Community

chaanks commited on Jul 17

Commit

ddab44e

•

1 Parent(s): 7902af3

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -59,10 +59,10 @@ hifi_gan_unit = UnitHIFIGAN.from_hparams(source="speechbrain/tts-hifigan-unit-hu
 # Running the S2UT model
 codes = s2ut.translate_file("speechbrain/s2st-transformer-fr-en-hubert-l6-k100-cvss/example-fr.wav")
-codes = torch.IntTensor(codes)
 # Running Vocoder (units-to-waveform)
-waveforms = hifi_gan_unit.decode_unit(codes)
 # Save the waverform
 torchaudio.save('example.wav',waveforms.squeeze(1), 16000)

 # Running the S2UT model
 codes = s2ut.translate_file("speechbrain/s2st-transformer-fr-en-hubert-l6-k100-cvss/example-fr.wav")
+codes = torch.IntTensor(codes[:-1])
 # Running Vocoder (units-to-waveform)
+waveforms = hifi_gan_unit.decode_unit(codes.unsqueeze(-1))
 # Save the waverform
 torchaudio.save('example.wav',waveforms.squeeze(1), 16000)