Update README.md

README.md

@@ -47,6 +47,7 @@ apt get install ffmpeg
 
 ```py
 import torchaudio
+
 from audiocraft.models import MusicGen
 from audiocraft.data.audio import audio_write
 
@@ -57,7 +58,6 @@ descriptions = ['happy rock', 'energetic EDM', 'sad jazz']
 
 wav = model.generate(descriptions)  # generates 3 samples.
 
-wav = model.generate_with_chroma(descriptions, melody[None].expand(3, -1, -1), sr)
 for idx, one_wav in enumerate(wav):
     # Will save under {idx}.wav, with loudness normalization at -14 db LUFS.
     audio_write(f'{idx}', one_wav.cpu(), model.sample_rate, strategy="loudness")