Upload

app.py

@@ -88,19 +88,12 @@ def enhance(filepath):
 
 
 
-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",
-    cache_examples=False,
-)
-
-if __name__ == "__main__":
-    se_demo.launch()
+with gr.Blocks() as demo:
+    gr.Markdown(ABOUT)
+    input_audio = gr.Audio(label="Input Audio", type="filepath")
+    run_button = gr.Button("Enhance")
+    output_audio = gr.Audio(label="Enhanced Audio", type="filepath")
+
+    run_button.click(enhance, inputs=[input_audio], outputs=[output_audio])
+
+demo.queue().launch()