Update README.md

README.md

@@ -40,12 +40,8 @@ const audio_values = await model.generate({
 });
 
 // (Optional) Write the output to a WAV file
-import wavefile from 'wavefile';
-import fs from 'fs';
-
-const wav = new wavefile.WaveFile();
-wav.fromScratch(1, model.config.audio_encoder.sampling_rate, '32f', audio_values.data);
-fs.writeFileSync('musicgen.wav', wav.toBuffer());
+const audio = new RawAudio(audio_values.data, model.config.audio_encoder.sampling_rate);
+audio.save('musicgen.wav');
 ```
 
 <audio controls src="https://cdn-uploads.huggingface.co/production/uploads/61b253b7ac5ecaae3d1efe0c/cJ9HFIDstOqJN0eGyJwlS.wav"></audio>