Update README.md

README.md

@@ -68,7 +68,7 @@ The resulting waveform can be saved as a `.wav` file:
 ```python
 import scipy
 
-scipy.io.wavfile.write("techno.wav", rate=model.config.sampling_rate, data=output)
+scipy.io.wavfile.write("techno.wav", rate=model.config.sampling_rate, data=output.float().numpy())
 ```
 
 Or displayed in a Jupyter Notebook / Google Colab:
@@ -76,7 +76,7 @@ Or displayed in a Jupyter Notebook / Google Colab:
 ```python
 from IPython.display import Audio
 
-Audio(output, rate=model.config.sampling_rate)
+Audio(output.numpy(), rate=model.config.sampling_rate)
 ```