inOXcrm commited on
Commit
3fbcf42
1 Parent(s): 2df30be

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -31,14 +31,14 @@ Use with Nemo Toolkit
31
  model = HifiGanModel.from_pretrained(model_name="tts_de_hui_hifigan_ft_fastpitch_multispeaker_5")
32
 
33
  # Generate audio
34
- import soundfile as sf
35
  parsed = spec_generator.parse("")
36
  speaker_id = 0
37
  spectrogram = spec_generator.generate_spectrogram(tokens=parsed, speaker=speaker_id)
38
  audio = model.convert_spectrogram_to_audio(spec=spectrogram)
39
 
40
  # Save the audio to disk in a file called speech.wav
41
- sf.write("speech.wav", audio.to('cpu').numpy(), 44100)
42
  ```
43
 
44
 
 
31
  model = HifiGanModel.from_pretrained(model_name="tts_de_hui_hifigan_ft_fastpitch_multispeaker_5")
32
 
33
  # Generate audio
34
+ import torchaudio
35
  parsed = spec_generator.parse("")
36
  speaker_id = 0
37
  spectrogram = spec_generator.generate_spectrogram(tokens=parsed, speaker=speaker_id)
38
  audio = model.convert_spectrogram_to_audio(spec=spectrogram)
39
 
40
  # Save the audio to disk in a file called speech.wav
41
+ torchaudio.save('german_sound.wav', audio.cpu(), 44100)
42
  ```
43
 
44