Upload

app.py

@@ -2,6 +2,7 @@ import shlex
 import subprocess
 import spaces
 import torch
+import gradio as gr
 
 # install packages for mamba
 def install_mamba():
@@ -12,8 +13,13 @@ def install_mamba():
 
 install_mamba()
 
+ABOUT = """
+# SEMamba: Speech Enhancement
+A Mamba-based model that denoises real-world audio.
+Upload or record a noisy clip and click **Enhance** to hear + see its spectrogram.
+"""
+
 
-import gradio as gr
 import torch
 import yaml
 import librosa
@@ -48,6 +54,7 @@ model.load_state_dict(sdict["generator"])
 model.eval()
 
 
+
    
 @spaces.GPU
 def enhance(filepath):
@@ -77,17 +84,13 @@ def enhance(filepath):
     librosa.display.specshow(S, sr=orig_sr, hop_length=512, x_axis="time", y_axis="hz", ax=ax)
     ax.set_title("Enhanced Spectrogram")
     plt.colorbar(format="%+2.0f dB", ax=ax)
+
     return out, fig
 
-ABOUT = """
-# SEMamba: Speech Enhancement
-A Mamba-based model that denoises real-world audio.
-Upload or record a noisy clip and click **Enhance** to hear + see its spectrogram.
-"""
 
 with gr.Blocks() as demo:
     gr.Markdown(ABOUT)
-    audio_in     = gr.Audio(label="Your Noisy Audio",
+    audio_in     = gr.Audio(label="Audio",
                             type="filepath",
                             interactive=True)
     run_button   = gr.Button("Enhance", variant="primary")