Update README.md

README.md

@@ -20,28 +20,27 @@ pip install vocos-mlx
 ```python
 from vocos_mlx import Vocos, load_audio, log_mel_spectrogram
 
-audio = load_audio("audio.wav", 24_000)
-mel_spec = log_mel_spectrogram(audio, n_mels = 100)
-
 vocos = Vocos.from_pretrained("lucasnewman/vocos-mel-24khz")
 
 # reconstruct
+audio = load_audio("audio.wav", 24_000)
 reconstructed_audio = vocos(audio)
 
 # decode from mel spec
-decoded_audio = vocos.decode(mx.expand_dims(mel_spec, axis=0))
+mel_spec = log_mel_spectrogram(audio, n_mels = 100)
+decoded_audio = vocos.decode(mel_spec)
 ```
+
 ### Encodec
 
 ```python
 from vocos_mlx import Vocos, load_audio
 
-audio = load_audio("audio.wav", 24_000)
-
 vocos = Vocos.from_pretrained("lucasnewman/vocos-encodec-24khz")
 
 # reconstruct
-reconstructed = vocos(audio, bandwidth_id = 3)
+audio = load_audio("audio.wav", 24_000)
+reconstructed_audio = vocos(audio, bandwidth_id = 3)
 
 # decode with encodec codes
 codes = ...