This commit is contained in:
adamnsandle
2024-06-28 14:13:18 +00:00
parent 5cd2ba54db
commit 136cdcdf5b

View File

@@ -37,6 +37,7 @@ def silero_vad(onnx=False, force_onnx_cpu=False):
raise Exception(f'Please install torch {supported_version} or greater ({installed_version} installed)')
model_dir = os.path.join(os.path.dirname(__file__), 'files')
print(model_dir, os.path.dirname(__file__))
if onnx:
model = OnnxWrapper(os.path.join(model_dir, 'silero_vad.onnx'), force_onnx_cpu)
else: