Update README.MD
#1
by faisaljan - opened
README.md
CHANGED
|
@@ -67,7 +67,9 @@ The resulting waveform can be saved as a `.wav` file:
|
|
| 67 |
|
| 68 |
```python
|
| 69 |
import scipy
|
|
|
|
| 70 |
|
|
|
|
| 71 |
scipy.io.wavfile.write("techno.wav", rate=model.config.sampling_rate, data=output)
|
| 72 |
```
|
| 73 |
|
|
|
|
| 67 |
|
| 68 |
```python
|
| 69 |
import scipy
|
| 70 |
+
import numpy as np
|
| 71 |
|
| 72 |
+
output_np = output.squeeze().detach().cpu().numpy()
|
| 73 |
scipy.io.wavfile.write("techno.wav", rate=model.config.sampling_rate, data=output)
|
| 74 |
```
|
| 75 |
|