remove debug stats vad
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user