Update webui for time_suffix

This commit is contained in:
litagin02
2024-03-13 16:51:23 +09:00
parent 1ddea1aa6d
commit aff666db75

View File

@@ -9,6 +9,7 @@ def do_slice(
min_sec: float, min_sec: float,
max_sec: float, max_sec: float,
min_silence_dur_ms: int, min_silence_dur_ms: int,
time_suffix: bool,
input_dir: str, input_dir: str,
): ):
if model_name == "": if model_name == "":
@@ -25,6 +26,8 @@ def do_slice(
"--min_silence_dur_ms", "--min_silence_dur_ms",
str(min_silence_dur_ms), str(min_silence_dur_ms),
] ]
if time_suffix:
cmd.append("--time_suffix")
if input_dir != "": if input_dir != "":
cmd += ["--input_dir", input_dir] cmd += ["--input_dir", input_dir]
# onnxの警告が出るので無視する # onnxの警告が出るので無視する
@@ -130,6 +133,10 @@ def create_dataset_app() -> gr.Blocks:
step=100, step=100,
label="無音とみなして区切る最小の無音の長さms", label="無音とみなして区切る最小の無音の長さms",
) )
time_suffix = gr.Checkbox(
value=False,
label="WAVファイル名の末尾に元ファイルの時間範囲を付与する",
)
slice_button = gr.Button("スライスを実行") slice_button = gr.Button("スライスを実行")
result1 = gr.Textbox(label="結果") result1 = gr.Textbox(label="結果")
with gr.Row(): with gr.Row():
@@ -172,7 +179,14 @@ def create_dataset_app() -> gr.Blocks:
result2 = gr.Textbox(label="結果") result2 = gr.Textbox(label="結果")
slice_button.click( slice_button.click(
do_slice, do_slice,
inputs=[model_name, min_sec, max_sec, min_silence_dur_ms, input_dir], inputs=[
model_name,
min_sec,
max_sec,
min_silence_dur_ms,
time_suffix,
input_dir,
],
outputs=[result1], outputs=[result1],
) )
transcribe_button.click( transcribe_button.click(