sanchit-gandhi HF staff commited on
Commit
c71de0f
1 Parent(s): 28cedf1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -68,7 +68,7 @@ The resulting waveform can be saved as a `.wav` file:
68
  ```python
69
  import scipy
70
 
71
- scipy.io.wavfile.write("techno.wav", rate=model.config.sampling_rate, data=output)
72
  ```
73
 
74
  Or displayed in a Jupyter Notebook / Google Colab:
@@ -76,7 +76,7 @@ Or displayed in a Jupyter Notebook / Google Colab:
76
  ```python
77
  from IPython.display import Audio
78
 
79
- Audio(output, rate=model.config.sampling_rate)
80
  ```
81
 
82
 
 
68
  ```python
69
  import scipy
70
 
71
+ scipy.io.wavfile.write("techno.wav", rate=model.config.sampling_rate, data=output.float().numpy())
72
  ```
73
 
74
  Or displayed in a Jupyter Notebook / Google Colab:
 
76
  ```python
77
  from IPython.display import Audio
78
 
79
+ Audio(output.numpy(), rate=model.config.sampling_rate)
80
  ```
81
 
82