Revert "Test pypi upload"

This commit is contained in:
makaveli
2024-02-08 12:23:15 +05:30
committed by GitHub
parent f98e309663
commit 6071cc1cc5
10 changed files with 46 additions and 325 deletions
+5
View File
@@ -10,7 +10,9 @@ 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
@@ -18,11 +20,13 @@ 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]
@@ -106,6 +110,7 @@ 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: