update

Dockerfile

@@ -14,8 +14,12 @@ ENV PATH="/code/vad_go_venv/bin:$PATH"
 RUN pip install --upgrade pip
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
+RUN bash install.sh
+
 RUN bash build_vad_go.sh
 
+ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/code/thirdparty/onnxruntime-linux-x64-1.20.1/lib
+
 USER root
 
 RUN chmod -R 777 .

install.sh

@@ -0,0 +1,73 @@
+#!/usr/bin/env bash
+
+# bash install.sh --stage 0 --stop_stage 2
+
+verbose=true;
+stage=-1
+stop_stage=0
+
+
+# parse options
+while true; do
+  [ -z "${1:-}" ] && break;  # break if there are no arguments
+  case "$1" in
+    --*) name=$(echo "$1" | sed s/^--// | sed s/-/_/g);
+      eval '[ -z "${'"$name"'+xxx}" ]' && echo "$0: invalid option $1" 1>&2 && exit 1;
+      old_value="(eval echo \\$$name)";
+      if [ "${old_value}" == "true" ] || [ "${old_value}" == "false" ]; then
+        was_bool=true;
+      else
+        was_bool=false;
+      fi
+
+      # Set the variable to the right value-- the escaped quotes make it work if
+      # the option had spaces, like --cmd "queue.pl -sync y"
+      eval "${name}=\"$2\"";
+
+      # Check that Boolean-valued arguments are really Boolean.
+      if $was_bool && [[ "$2" != "true" && "$2" != "false" ]]; then
+        echo "$0: expected \"true\" or \"false\": $1 $2" 1>&2
+        exit 1;
+      fi
+      shift 2;
+      ;;
+
+    *) break;
+  esac
+done
+
+
+work_dir="$(pwd)"
+thirdparty_dir="$(pwd)/thirdparty"
+pretrained_models_dir="$(pwd)/pretrained_models"
+silero_vad_model_dir="${pretrained_models_dir}/silero_vad"
+
+mkdir -p "${thirdparty_dir}"
+mkdir -p "${pretrained_models_dir}"
+mkdir -p "${silero_vad_model_dir}"
+
+
+if [ ${stage} -le 0 ] && [ ${stop_stage} -ge 0 ]; then
+  $verbose && echo "stage 0: install onnxruntime"
+
+  cd "${thirdparty_dir}"
+
+  # install onnxruntime
+  wget https://github.com/microsoft/onnxruntime/releases/download/v1.20.1/onnxruntime-linux-x64-1.20.1.tgz
+  tar -zxvf onnxruntime-linux-x64-1.20.1.tgz
+  export C_INCLUDE_PATH=$C_INCLUDE_PATH:$(pwd)/onnxruntime-linux-x64-1.20.1/include
+  export LIBRARY_PATH=$LIBRARY_PATH:$(pwd)/onnxruntime-linux-x64-1.20.1/lib
+
+fi
+
+
+if [ ${stage} -le 1 ] && [ ${stop_stage} -ge 1 ]; then
+  $verbose && echo "stage 1: download silero_vad.onnx"
+
+  cd "${silero_vad_model_dir}"
+
+  # download model
+  # https://github.com/snakers4/silero-vad/tree/master/src/silero_vad/data
+  wget https://github.com/snakers4/silero-vad/raw/refs/heads/master/src/silero_vad/data/silero_vad.onnx
+
+fi

main.py

@@ -106,7 +106,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")
-                upload_button = gr.Button("Run VAD")
+                upload_button = gr.Button("Run VAD", variant="primary")
                 uploaded_output = gr.Textbox(label="outputs")
 
                 gr.Examples(
@@ -137,7 +137,7 @@ def main():
                 )
             with gr.TabItem("shell"):
                 shell_text = gr.Textbox(label="cmd")
-                shell_button = gr.Button("run")
+                shell_button = gr.Button("run", variant="primary")
                 shell_output = gr.Textbox(label="output")
 
                 shell_button.click(

silero/build_silero_vad.sh

@@ -1,19 +1,6 @@
 #!/usr/bin/env bash
 # https://github.com/snakers4/silero-vad/tree/master/examples/go
 
-# install onnxruntime
-wget https://github.com/microsoft/onnxruntime/releases/download/v1.20.1/onnxruntime-linux-x64-1.20.1.tgz
-tar -zxvf onnxruntime-linux-x64-1.20.1.tgz
-export C_INCLUDE_PATH=$C_INCLUDE_PATH:$(pwd)/onnxruntime-linux-x64-1.20.1/include
-export LIBRARY_PATH=$LIBRARY_PATH:$(pwd)/onnxruntime-linux-x64-1.20.1/lib
-#export LD_LIBRARY_PATH=$(pwd)/onnxruntime-linux-x64-1.20.1/lib:$LD_LIBRARY_PATH
-cp $(pwd)/onnxruntime-linux-x64-1.20.1/lib/* /usr/local/lib/
-
-# download model
-# https://github.com/snakers4/silero-vad/tree/master/src/silero_vad/data
-wget https://github.com/snakers4/silero-vad/raw/refs/heads/master/src/silero_vad/data/silero_vad.onnx
-
-# build
 mkdir build
 
 go env -w GO111MODULE=on

silero/main.go

@@ -11,7 +11,7 @@ import (
 
 func main() {
 	sd, err := speech.NewDetector(speech.DetectorConfig{
-		ModelPath:            "./silero/silero_vad.onnx",
+		ModelPath:            "./pretrained_models/silero_vad/silero_vad.onnx",
 		SampleRate:           8000,
 		Threshold:            0.5,
 		MinSilenceDurationMs: 100,