google
/

music-spectrogram-diffusion

SpectrogramDiffusionPipeline

Model card Files Files and versions Community

kashif HF staff commited on Mar 21, 2023

Commit

fd6d2e9

·

1 Parent(s): 6786dba

Update README.md

Files changed (1) hide show

README.md +2 -3

README.md CHANGED Viewed

@@ -25,12 +25,11 @@ As depicted above the model takes as input a MIDI file and tokenizes it into a s
 ```python
 from diffusers import SpectrogramDiffusionPipeline, MidiProcessor
-pipe = SpectrogramDiffusionPipeline.from_pretrained("google/music-spectrogram-diffusion")
 pipe = pipe.to("cuda")
-# Download via: wget http://www.piano-midi.de/midis/beethoven/beethoven_hammerklavier_2.mid
 processor = MidiProcessor()
 output = pipe(processor("beethoven_hammerklavier_2.mid"))
 audio = output.audios[0]

 ```python
 from diffusers import SpectrogramDiffusionPipeline, MidiProcessor
+pipe = SpectrogramDiffusionPipeline.from_pretrained("kashif/music-spectrogram-diffusion")
 pipe = pipe.to("cuda")
 processor = MidiProcessor()
+# Download MIDI from: wget http://www.piano-midi.de/midis/beethoven/beethoven_hammerklavier_2.mid
 output = pipe(processor("beethoven_hammerklavier_2.mid"))
 audio = output.audios[0]