add vad model to suppress warnings

2023-09-27 15:43:28 +08:00
parent e89afcc40f
commit 581406e5ab
2 changed files with 114 additions and 5 deletions
@@ -17,6 +17,7 @@ import torch
 import numpy as np
 import time
 from whisper_live.transcriber import WhisperModel
+from whisper_live.vad import VoiceActivityDetection


 class TranscriptionServer:
@@ -29,11 +30,7 @@ class TranscriptionServer:
    RATE = 16000
    def __init__(self):
        # voice activity detection model
-        self.vad_model, _ = torch.hub.load(repo_or_dir='snakers4/silero-vad',
-                                           model='silero_vad',
-                                           force_reload=True,
-                                           onnx=True
-                                           )
+        self.vad_model = VoiceActivityDetection()
        self.vad_threshold = 0.4
        self.clients = {}
        self.websockets = {}