Adel-Moumen commited on
Commit
ac3fca3
1 Parent(s): e7ee96b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -40,7 +40,7 @@ Please notice that we encourage you to read our tutorials and learn more about
40
  - *Basic Usage:*
41
  ```python
42
  import torch
43
- from speechbrain.inference.TTS import HIFIGAN
44
  hifi_gan = HIFIGAN.from_hparams(source="speechbrain/tts-hifigan-ljspeech", savedir="tmpdir")
45
  mel_specs = torch.rand(2, 80,298)
46
  waveforms = hifi_gan.decode_batch(mel_specs)
@@ -50,7 +50,7 @@ waveforms = hifi_gan.decode_batch(mel_specs)
50
 
51
  ```python
52
  import torchaudio
53
- from speechbrain.inference.TTS import HIFIGAN
54
  from speechbrain.lobes.models.FastSpeech2 import mel_spectogram
55
 
56
  # Load a pretrained HIFIGAN Vocoder
@@ -93,8 +93,8 @@ torchaudio.save('waveform_reconstructed.wav', waveforms.squeeze(1), 22050)
93
  ### Using the Vocoder with the TTS
94
  ```python
95
  import torchaudio
96
- from speechbrain.inference.TTS import Tacotron2
97
- from speechbrain.inference.TTS import HIFIGAN
98
 
99
  # Intialize TTS (tacotron2) and Vocoder (HiFIGAN)
100
  tacotron2 = Tacotron2.from_hparams(source="speechbrain/tts-tacotron2-ljspeech", savedir="tmpdir_tts")
40
  - *Basic Usage:*
41
  ```python
42
  import torch
43
+ from speechbrain.inference.vocoders import HIFIGAN
44
  hifi_gan = HIFIGAN.from_hparams(source="speechbrain/tts-hifigan-ljspeech", savedir="tmpdir")
45
  mel_specs = torch.rand(2, 80,298)
46
  waveforms = hifi_gan.decode_batch(mel_specs)
50
 
51
  ```python
52
  import torchaudio
53
+ from speechbrain.inference.vocoders import HIFIGAN
54
  from speechbrain.lobes.models.FastSpeech2 import mel_spectogram
55
 
56
  # Load a pretrained HIFIGAN Vocoder
93
  ### Using the Vocoder with the TTS
94
  ```python
95
  import torchaudio
96
+ from speechbrain.inference.vocoders import Tacotron2
97
+ from speechbrain.inference.vocoders import HIFIGAN
98
 
99
  # Intialize TTS (tacotron2) and Vocoder (HiFIGAN)
100
  tacotron2 = Tacotron2.from_hparams(source="speechbrain/tts-tacotron2-ljspeech", savedir="tmpdir_tts")