Upload

app.py

@@ -87,16 +87,20 @@ def enhance(filepath):
 
     return out, fig
 
-
-with gr.Blocks() as demo:
-    gr.Markdown(ABOUT)
-    audio_in     = gr.Audio(label="Audio",
-                            type="filepath",
-                            interactive=True)
-    run_button   = gr.Button("Enhance", variant="primary")
-    enhanced_out = gr.Audio(label="Enhanced Audio", interactive=False)
-    spec_out     = gr.Plot(label="Spectrogram")
-
-    run_button.click(enhance, inputs=audio_in, outputs=[enhanced_out, spec_out])
-
-demo.queue().launch()
+se_demo = gr.Interface(
+    fn = enhance,
+    inputs = [
+        gr.Audio(label="Input Audio", type="filepath"),
+        gr.Checkbox(label="Apply Speech Enhancement", value=True),
+    ],
+    outputs = [
+        gr.Audio(label="Output Audio", type="filepath"),
+        gr.Plot(label="Spectrogram")
+    ],
+    title = "SEMamba",
+)
+
+with demo:
+    se_demo
+
+demo.launch()