fix missing import

#2
by pete-rrr - opened
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")