Update app.py

app.py

@@ -1,27 +1,14 @@
-import librosa
-import librosa.display
-import PIL
 import gradio as gr
 import numpy as np
-import matplotlib.pyplot as plt
+import librosa
+
+# import the librosa library for converting audio to a Mel spectrogram image
 
-def scale_minmax(X, min=0.0, max=1.0):
-    X_std = (X - X.min()) / (X.max() - X.min())
-    X_scaled = X_std * (max - min) + min
-    return X_scaled
-def greet(audio):
-    path = audio
-    y, sr = librosa.load(path)
-    print(path)
-    np.clip(y,-1,1,out=y)
-    print(y)
-    
-    mel = librosa.feature.melspectrogram(y=y, sr=sr)   
-     # #https://stackoverflow.com/questions/56719138/how-can-i-save-a-librosa-spectrogram-plot-as-a-specific-sized-image/57204349#57204349
-    mel = np.log(mel + 1e-9)  
-    img = scale_minmax(mel, 0, 255).astype(np.uint8)
-    img = 255-img 
-    return img
+def audio_to_mel(audio):
+  # convert the audio to a Mel spectrogram image using librosa
+  mel_spectrogram = librosa.feature.melspectrogram(audio)
+  return mel_spectrogram
 
-iface = gr.Interface(fn=greet, inputs= gr.Audio(source="upload",type="filepath"), outputs=gr.Image(type="pil",label="Mel spectrogram"))
-iface.launch()
+# create the gradio app
+app = gr.Interface(audio_to_mel, "audio", "image")
+app.launch()