Add split audio for batch

IAHispano · Jan 31, 2024 · 5576bcc · 5576bcc
1 parent 37d091a
commit 5576bcc
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/core.py b/core.py
@@ -72,6 +72,7 @@ def run_batch_infer_script(
     output_folder,
     pth_file,
     index_path,
+    split_audio,
 ):
     infer_script_path = os.path.join("rvc", "infer", "infer.py")
 
@@ -104,6 +105,7 @@ def run_batch_infer_script(
             output_path,
             pth_file,
             index_path,
+            str(split_audio),
         ]
         subprocess.run(command)
 

diff --git a/tabs/inference/inference.py b/tabs/inference/inference.py
@@ -351,6 +351,12 @@ def inference_tab():
                 clear_outputs = gr.Button(
                     i18n("Clear Outputs (Deletes all audios in assets/audios)")
                 )
+                split_audio_batch = gr.Checkbox(
+                    label=i18n("Split Audio"),
+                    visible=True,
+                    value=False,
+                    interactive=True,
+                )
                 pitch_batch = gr.Slider(
                     minimum=-24,
                     maximum=24,
@@ -460,6 +466,7 @@ def toggle_visible(checkbox):
             output_folder_batch,
             model_file,
             index_file,
+            split_audio_batch,
         ],
         outputs=[vc_output3],
     )