Update app.py

app.py

@@ -2,14 +2,17 @@ import gradio as gr
 import numpy as np
 import librosa
 
-# import the librosa library for converting audio to a Mel spectrogram image
+# import the librosa library for converting a Mel spectrogram image to audio
 
-def audio_to_mel(audio):
-  # convert the audio to a Mel spectrogram image using librosa
+def mel_to_audio(mel_spectrogram):
+  # convert the Mel spectrogram image to a grayscale image to avoid the string indices must be integers error
+  mel_spectrogram = np.mean(mel_spectrogram, axis=0)
+
+  # convert the Mel spectrogram image to audio using librosa
   # specify the data type of the output array to avoid the same_kind casting error
-  mel_spectrogram = librosa.feature.melspectrogram(audio, dtype=np.float64)
-  return mel_spectrogram
+  audio = librosa.feature.inverse.mel_to_audio(mel_spectrogram, dtype=np.float64)
+  return audio
 
 # create the gradio app
-app = gr.Interface(audio_to_mel, "audio", "image")
+app = gr.Interface(mel_to_audio, "image", "audio")
 app.launch()