Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import matplotlib.pyplot as plt
+from matplotlib.colors import LinearSegmentedColormap
 import numpy as np
 import os
 import soundfile as sf
@@ -46,6 +47,19 @@ def main():
 
 def create_spectrogram_and_get_info(audio_file):
     plt.clf()
+    
+    cdict = {'red':   [(0.0,  0.5, 0.5),
+                       (0.5,  1.0, 1.0),
+                       (1.0,  1.0, 1.0)],
+             'green': [(0.0,  0.0, 0.0),
+                       (0.25, 0.0, 0.0),
+                       (0.75, 1.0, 1.0),
+                       (1.0,  1.0, 1.0)],
+             'blue':  [(0.0,  0.5, 0.5),
+                       (0.5,  0.0, 0.0),
+                       (1.0,  0.0, 0.0)]}
+    custom_cmap = LinearSegmentedColormap('CustomMap', cdict)
+    
     plt.figure(figsize=(15, 5))
     audio_data, sample_rate = sf.read(audio_file)
     if len(audio_data.shape) > 1: