diff --git a/whisper_live/vad.py b/whisper_live/vad.py index 53170f9..31edac8 100644 --- a/whisper_live/vad.py +++ b/whisper_live/vad.py @@ -10,9 +10,7 @@ import onnxruntime class VoiceActivityDetection(): def __init__(self, force_onnx_cpu=True): - print("downloading ONNX model...") path = self.download() - print("loading session") opts = onnxruntime.SessionOptions() opts.log_severity_level = 3 @@ -20,13 +18,11 @@ class VoiceActivityDetection(): opts.inter_op_num_threads = 1 opts.intra_op_num_threads = 1 - print("loading onnx model") if force_onnx_cpu and 'CPUExecutionProvider' in onnxruntime.get_available_providers(): self.session = onnxruntime.InferenceSession(path, providers=['CPUExecutionProvider'], sess_options=opts) else: self.session = onnxruntime.InferenceSession(path, providers=['CUDAExecutionProvider'], sess_options=opts) - print("reset states") self.reset_states() self.sample_rates = [8000, 16000] @@ -110,7 +106,6 @@ class VoiceActivityDetection(): # Check if the model file already exists if not os.path.exists(model_filename): # If it doesn't exist, download the model using wget - print("Downloading VAD ONNX model...") try: subprocess.run(["wget", "-O", model_filename, model_url], check=True) except subprocess.CalledProcessError: