pete-rrr commited on
Commit
82db64e
1 Parent(s): 6b1b5aa

fix missing import

Browse files

add datasets import to code example

Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -33,6 +33,7 @@ Use the code below to convert text into a mono 16 kHz speech waveform.
33
  from transformers import SpeechT5Processor, SpeechT5ForTextToSpeech, SpeechT5HifiGan
34
  import torch
35
  import soundfile as sf
 
36
 
37
  processor = SpeechT5Processor.from_pretrained("microsoft/speecht5_tts")
38
  model = SpeechT5ForTextToSpeech.from_pretrained("microsoft/speecht5_tts")
 
33
  from transformers import SpeechT5Processor, SpeechT5ForTextToSpeech, SpeechT5HifiGan
34
  import torch
35
  import soundfile as sf
36
+ from datasets import load_dataset
37
 
38
  processor = SpeechT5Processor.from_pretrained("microsoft/speecht5_tts")
39
  model = SpeechT5ForTextToSpeech.from_pretrained("microsoft/speecht5_tts")