Upload

app.py

@@ -81,16 +81,19 @@ def enhance(audio, do_pcs):
 
     return orig_sr, out
 
-demo = gr.Interface(
-    fn=enhance,
-    inputs=[
-        gr.Audio(source="upload", type="numpy", label="Noisy wav"),
-        gr.Checkbox(label="Apply PCS post-processing", value=False),
-    ],
-    outputs=gr.Audio(type="numpy", label="Enhanced wav"),
-    title="SEMamba Speech Enhancement",
-    description="Upload a noisy WAV; tick **Apply PCS** for the pcs400 filter.",
-)
+with gr.Blocks() as demo:
+    gr.Markdown("## SEMamba Speech Enhancement demo")
+    with gr.Row():
+        upload = gr.Audio(label="Upload WAV", type="numpy")
+        record = gr.Audio(label="Record via mic", type="numpy")
+    pcs    = gr.Checkbox(label="Apply PCS post-processing", value=False)
+    btn    = gr.Button("Enhance")
+    out    = gr.Audio(label="Enhanced WAV", type="numpy")
+
+    def runner(up, rec, do_pcs):
+        return enhance(up if up else rec, do_pcs)
+
+    btn.click(runner, [upload, record, pcs], out)
 
 
 demo.launch()