feat: add fast checkbox

app.py

@@ -3,11 +3,14 @@ import gradio as gr
 from scipy.io.wavfile import write
 
 
-def inference(audio):
+def inference(audio, is_fast):
     os.makedirs("out", exist_ok=True)
     write('test.wav', audio[0], audio[1])
     os.environ['DANNA_CHECKPOINTS'] = './checkpoints'
-    os.system("danna_sep --outdir out test.wav")
+    cmd = "danna_sep --outdir out test.wav"
+    if is_fast:
+        cmd += " --fast"
+    os.system(cmd)
     return "./out/test_vocals.wav", "./out/test_bass.wav",\
         "./out/test_drums.wav", "./out/test_other.wav"
 
@@ -19,7 +22,10 @@ article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2112.037
 examples = []
 gr.Interface(
     inference,
-    gr.inputs.Audio(type="numpy", label="Input"),
+    [
+        gr.inputs.Audio(type="numpy", label="Input"),
+        gr.inputs.Checkbox(label="Faster inference without demucs")
+    ],
     [gr.outputs.Audio(type="file", label="Vocals"), gr.outputs.Audio(type="file", label="Bass"), gr.outputs.Audio(
         type="file", label="Drums"), gr.outputs.Audio(type="file", label="Other")],
     title=title,