Update README.md

README.md

@@ -40,7 +40,7 @@ Please notice that we encourage you to read our tutorials and learn more about
 import torch
 from speechbrain.inference.vocoders import UnitHIFIGAN
 
-hifi_gan_unit = UnitHIFIGAN.from_hparams(source="speechbrain/tts-hifigan-unit-hubert-l6-k100-ljspeech", savedir="tmpdir_vocoder")
+hifi_gan_unit = UnitHIFIGAN.from_hparams(source="speechbrain/tts-hifigan-unit-hubert-l6-k100-ljspeech", savedir="pretrained_models/tts-hifigan-unit-hubert-l6-k100-ljspeech")
 codes = torch.randint(0, 99, (100,))
 waveform = hifi_gan_unit.decode_unit(codes)
 
@@ -54,8 +54,8 @@ from speechbrain.inference.ST import EncoderDecoderS2UT
 from speechbrain.inference.vocoders import UnitHIFIGAN
 
 # Intialize S2UT (Transformer) and Vocoder (HiFIGAN Unit)
-s2ut = EncoderDecoderS2UT.from_hparams(source="speechbrain/s2st-transformer-fr-en-hubert-l6-k100-cvss", savedir="tmpdir_s2ut")
-hifi_gan_unit = UnitHIFIGAN.from_hparams(source="speechbrain/tts-hifigan-unit-hubert-l6-k100-ljspeech", savedir="tmpdir_vocoder")
+s2ut = EncoderDecoderS2UT.from_hparams(source="speechbrain/s2st-transformer-fr-en-hubert-l6-k100-cvss", savedir="pretrained_models/s2st-transformer-fr-en-hubert-l6-k100-cvss")
+hifi_gan_unit = UnitHIFIGAN.from_hparams(source="speechbrain/tts-hifigan-unit-hubert-l6-k100-ljspeech", savedir="tpretrained_models/tts-hifigan-unit-hubert-l6-k100-ljspeech")
 
 # Running the S2UT model
 codes = s2ut.translate_file("speechbrain/s2st-transformer-fr-en-hubert-l6-k100-cvss/example-fr.wav")