fix: ffmpeg does not overwrite the merged audio

app/infer.py

@@ -65,7 +65,7 @@ def merge(exp_dir: str, original_audio: str, vocal: Tuple[int, np.ndarray]) -> s
     sf.write(tmp, vocal[1], vocal[0])
 
     os.system(
-        f"ffmpeg -i {music} -i {tmp} -filter_complex '[1]volume=2[a];[0][a]amix=inputs=2:duration=first:dropout_transition=2' {tmp}.merged.mp3"
+        f"ffmpeg -i {music} -i {tmp} -filter_complex '[1]volume=2[a];[0][a]amix=inputs=2:duration=first:dropout_transition=2' -ac 2 -y {tmp}.merged.mp3"
     )
 
     return f"{tmp}.merged.mp3"
@@ -91,7 +91,7 @@ class InferenceTab:
 
             with gr.Column():
                 self.f0add = gr.Slider(
-                    label="F0 add",
+                    label="F0 +/-",
                     minimum=-16,
                     maximum=16,
                     step=1,
@@ -115,11 +115,11 @@ class InferenceTab:
             self.infer_btn = gr.Button(value="Infer", variant="primary")
         with gr.Row():
             self.infer_output = gr.Audio(
-                label="Inferred audio", show_download_button=True
+                label="Inferred audio", show_download_button=True, format="mp3"
             )
         with gr.Row():
             self.merge_output = gr.Audio(
-                label="Merged audio", show_download_button=True
+                label="Merged audio", show_download_button=True, format="mp3"
             )
 
     def build(self, exp_dir: gr.Textbox):

app/train.py

@@ -162,7 +162,7 @@ class TrainTab:
         gr.Markdown("# Training")
         gr.Markdown(
             "You can start training the model by clicking the button below. "
-            "Each time you click the button, the model will train for 20 epochs, which takes about 10 minutes on ZeroGPU (A100). "
+            "Each time you click the button, the model will train for 10 epochs, which takes about 3 minutes on ZeroGPU (A100). "
             "Tha latest *training checkpoint* will be avaible below."
         )
 
@@ -178,10 +178,6 @@ class TrainTab:
             fn=train_model,
             inputs=[exp_dir],
             outputs=[self.latest_checkpoint],
-        ).success(
-            fn=train_model,
-            inputs=[exp_dir],
-            outputs=[self.latest_checkpoint],
         )
 
         self.train_index_btn.click(