update

README.md

@@ -13,12 +13,12 @@ Check out the configuration reference at https://huggingface.co/docs/hub/spaces-
 ### Vad GO
 
 ```text
-docker build -t vad_go:v1 .
+docker build -t vad_go:v20241211_1518 .
 
 docker run -itd \
 --name vad_go \
 --network host \
-vad_go:v1
+vad_go:v20241211_1518
 
 http://10.75.27.247:7860
 ```

silero_vad/main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"flag"
 	"fmt"
 	"log"
 	"os"
@@ -9,12 +10,27 @@ import (
 )
 
 func main() {
+    modelPathPtr := flag.String("model_path", "./pretrained_models/silero_vad/silero_vad.onnx", "silero vad onnx model")
+    filenamePtr := flag.String("filename", "", "input wav audio file")
+    silenceTimePtr := flag.Float64("silence_time", 0.1, "in the end of each speech chunk wait for min_silence_duration_ms before separating it")
+	speechPadTimePtr := flag.Float64("speech_pad_time", 0.03, "final speech chunks are padded by speech_pad_ms each side")
+	flag.Parse()
+
+    var modelPath string = *modelPathPtr
+    var filename string = *filenamePtr
+    var silenceTime float32 = float32(*silenceTimePtr)
+    var speechPadTime float32 = float32(*speechPadTimePtr)
+	fmt.Println(filename)
+
+    var silenceTimeMs uint32 = uint32(silenceTime * 1e3)
+    var speechPadTimeMs uint32 = uint32(speechPadTime * 1e3)
+
 	sd, err := speech.NewDetector(speech.DetectorConfig{
-		ModelPath:            "./pretrained_models/silero_vad/silero_vad.onnx",
+		ModelPath:            modelPath,
 		SampleRate:           8000,
 		Threshold:            0.5,
-		MinSilenceDurationMs: 100,
-		SpeechPadMs:          30,
+		MinSilenceDurationMs: silenceTimeMs,
+		SpeechPadMs:          speechPadTimeMs,
 	})
 	if err != nil {
 		log.Fatalf("failed to create speech detector: %s", err)