Update README.md

README.md

@@ -1,7 +1,17 @@
+---
+license: mit
+tags:
+- stable-diffusion
+- stable-diffusion-diffusers
+- text-to-image
+---
+
+[Dance Diffusion](https://github.com/Harmonai-org/sample-generator) is now available in 🧨 Diffusers.
+
 ## FP32
 
 ```python
-# !pip install git+https://github.com/huggingface/diffusers.git
+# !pip install diffusers[torch] accelerate scipy
 from diffusers import DiffusionPipeline
 import scipy
 
@@ -13,7 +23,7 @@ audios = pipeline(audio_length_in_s=4.0).audios
 
 # To save locally
 for audio in audios:
-    scipy.io.wavfile.write("maestro_test.wav", pipe.unet.sample_rate, audio.transpose())
+    scipy.io.wavfile.write(f"maestro_test_{i}.wav", pipe.unet.sample_rate, audio.transpose())
     
 # To dislay in google colab
 import IPython.display as ipd
@@ -26,7 +36,7 @@ for audio in audios:
 Faster at a small loss of quality
 
 ```python
-# !pip install git+https://github.com/huggingface/diffusers.git
+# !pip install diffusers[torch] accelerate scipy
 from diffusers import DiffusionPipeline
 import scipy
 import torch
@@ -38,8 +48,8 @@ pipeline = pipeline.to("cuda")
 audios = pipeline(audio_length_in_s=4.0).audios
 
 # To save locally
-for audio in audios:
-    scipy.io.wavfile.write("maestro_test.wav", pipe.unet.sample_rate, audio.transpose())
+for i, audio in enumerate(audios):
+    scipy.io.wavfile.write(f"maestro_test_{i}.wav", pipe.unet.sample_rate, audio.transpose())
     
 # To dislay in google colab
 import IPython.display as ipd