update

main.py

@@ -32,11 +32,17 @@ def get_args():
     return args
 
 
-def process_uploaded_file(vad_engine: str, filename: str, silence_time: float = 0.3, longest_activate: float = 3.0) -> Tuple[str, str]:
+def process_uploaded_file(
+        vad_engine: str,
+        filename: str,
+        silence_time: float = 0.3,
+        longest_activate: float = 3.0,
+        uploaded_threshold: float = 0.5,
+) -> Tuple[str, str]:
     if vad_engine == "nx_vad":
         return run_nx_vad(filename, silence_time, longest_activate)
     elif vad_engine == "silero_vad":
-        return run_silero_vad(filename, silence_time, longest_activate)
+        return run_silero_vad(filename, silence_time, uploaded_threshold)
     else:
         return f"vad engine invalid: {vad_engine}", ""
 
@@ -61,13 +67,13 @@ def run_nx_vad(filename: str, silence_time: float = 0.3, longest_activate: float
     return vad_timestamps, raw_vad_result
 
 
-def run_silero_vad(filename: str, silence_time: float = 0.3, longest_activate: float = 3.0) -> Tuple[str, str]:
+def run_silero_vad(filename: str, silence_time: float = 0.3, threshold: float = 0.5) -> Tuple[str, str]:
     filename = Path(filename).as_posix()
 
     main_logger.info("do silero vad: {}".format(filename))
 
-    cmd = "vad_bins/silero_vad --filename {} --silence_time {}".format(
-        filename, silence_time
+    cmd = "vad_bins/silero_vad --filename {} --silence_time {} --threshold {}".format(
+        filename, silence_time, threshold
     )
     raw_vad_result = Command.popen(cmd)
 
@@ -119,6 +125,7 @@ def main():
                     uploaded_vad_engine = gr.Dropdown(choices=["nx_vad", "silero_vad"], value="nx_vad", label="vad_engine")
                     uploaded_silence_time = gr.Slider(minimum=0.0, maximum=1.0, value=0.3, step=0.01, label="silence time")
                     uploaded_longest_activate = gr.Slider(minimum=0.0, maximum=20.0, value=3.0, step=0.1, label="longest activate")
+                    uploaded_threshold = gr.Slider(minimum=0.0, maximum=1.0, value=0.5, step=0.1, label="threshold")
                 upload_button = gr.Button("Run VAD", variant="primary")
 
                 with gr.Row():
@@ -132,6 +139,7 @@ def main():
                         uploaded_file,
                         uploaded_silence_time,
                         uploaded_longest_activate,
+                        uploaded_threshold,
                     ],
                     outputs=[
                         uploaded_vad_timestamps,
@@ -147,6 +155,7 @@ def main():
                         uploaded_file,
                         uploaded_silence_time,
                         uploaded_longest_activate,
+                        uploaded_threshold,
                     ],
                     outputs=[
                         uploaded_vad_timestamps,