feat: show download button on audio

app/infer.py

@@ -59,7 +59,7 @@ def infer(
 def merge(exp_dir: str, original_audio: str, vocal: Tuple[int, np.ndarray]) -> str:
     base = os.path.basename(original_audio)
     base = os.path.splitext(base)[0]
-    music = os.path.join("separated", "htdemucs", base, "no-vocals.wav")
+    music = os.path.join("separated", "htdemucs", base, "no_vocals.wav")
 
     tmp = os.path.join(exp_dir, "tmp.wav")
     sf.write(tmp, vocal[1], vocal[0])
@@ -83,12 +83,12 @@ class InferenceTab:
         )
 
         with gr.Row():
-            with gr.Column():
-                self.original_audio = gr.Audio(
-                    label="Upload original audio",
-                    type="filepath",
-                    show_download_button=True,
-                )
+            self.original_audio = gr.Audio(
+                label="Upload original audio",
+                type="filepath",
+                show_download_button=True,
+            )
+
             with gr.Column():
                 self.f0add = gr.Slider(
                     label="F0 add",
@@ -111,12 +111,16 @@ class InferenceTab:
                     step=0.01,
                     value=0.33,
                 )
-            with gr.Column():
-                self.infer_btn = gr.Button(value="Infer", variant="primary")
+
+            self.infer_btn = gr.Button(value="Infer", variant="primary")
         with gr.Row():
-            self.infer_output = gr.Audio(label="Inferred audio")
+            self.infer_output = gr.Audio(
+                label="Inferred audio", show_download_button=True
+            )
         with gr.Row():
-            self.merge_output = gr.Audio(label="Merged audio")
+            self.merge_output = gr.Audio(
+                label="Merged audio", show_download_button=True
+            )
 
     def build(self, exp_dir: gr.Textbox):
         self.infer_btn.click(