Update README.md

README.md

@@ -11,7 +11,7 @@ tags:
 ```python
 # !pip install diffusers[torch] accelerate scipy
 from diffusers import DiffusionPipeline
-import scipy.io
+from scipy.io.wavfile import write
 
 model_id = "harmonai/maestro-150k"
 pipeline = DiffusionPipeline.from_pretrained(model_id)
@@ -21,7 +21,7 @@ audios = pipeline(audio_length_in_s=4.0).audios
 
 # To save locally
 for i, audio in enumerate(audios):
-    scipy.io.wavfile.write(f"maestro_test_{i}.wav", pipe.unet.sample_rate, audio.transpose())
+    write(f"maestro_test_{i}.wav", pipe.unet.sample_rate, audio.transpose())
     
 # To dislay in google colab
 import IPython.display as ipd
@@ -36,7 +36,7 @@ Faster at a small loss of quality
 ```python
 # !pip install diffusers[torch] accelerate scipy
 from diffusers import DiffusionPipeline
-import scipy.io
+from scipy.io.wavfile import write
 import torch
 
 model_id = "harmonai/maestro-150k"
@@ -47,7 +47,7 @@ audios = pipeline(audio_length_in_s=4.0).audios
 
 # To save locally
 for i, audio in enumerate(audios):
-    scipy.io.wavfile.write(f"maestro_test_{i}.wav", pipe.unet.sample_rate, audio.transpose())
+    write(f"maestro_test_{i}.wav", pipe.unet.sample_rate, audio.transpose())
     
 # To dislay in google colab
 import IPython.display as ipd