Added function to save the audio files

README.md

@@ -70,6 +70,19 @@ import scipy
 
 scipy.io.wavfile.write("techno.wav", rate=model.config.sampling_rate, data=output.float().numpy())
 ```
+If you find any error while saving the audio file using the above code then you can try the following code
+
+```python
+def save_audio_file(output, file):
+    audio_array = output.numpy().squeeze()
+    audio_array /=1.414
+    audio_array *= 32767
+    audio_array = audio_array.astype(np.int16)
+
+    sampling_rate = model.config.sampling_rate
+
+    scipy.io.wavfile.write(f"/root/{file}", rate=sampling_rate, data=audio_array)
+```
 
 Or displayed in a Jupyter Notebook / Google Colab: