1
1
dependencies = ['torch' , 'torchaudio' ]
2
2
import torch
3
+ import os
3
4
import json
4
5
from utils_vad import (init_jit_model ,
5
6
get_speech_timestamps ,
@@ -20,11 +21,11 @@ def silero_vad(onnx=False):
20
21
Returns a model with a set of utils
21
22
Please see https://github.com/snakers4/silero-vad for usage examples
22
23
"""
23
- hub_dir = torch . hub . get_dir ( )
24
+ model_dir = os . path . join ( os . path . dirname ( __file__ ), 'files' )
24
25
if onnx :
25
- model = OnnxWrapper (f' { hub_dir } /snakers4_silero-vad_master/files/ silero_vad.onnx' )
26
+ model = OnnxWrapper (os . path . join ( model_dir , ' silero_vad.onnx') )
26
27
else :
27
- model = init_jit_model (model_path = f' { hub_dir } /snakers4_silero-vad_master/files/ silero_vad.jit' )
28
+ model = init_jit_model (os . path . join ( model_dir , ' silero_vad.jit') )
28
29
utils = (get_speech_timestamps ,
29
30
save_audio ,
30
31
read_audio ,
@@ -75,17 +76,17 @@ def silero_lang_detector_95(onnx=False):
75
76
Please see https://github.com/snakers4/silero-vad for usage examples
76
77
"""
77
78
78
- hub_dir = torch .hub .get_dir ()
79
79
if onnx :
80
80
url = 'https://models.silero.ai/vad_models/lang_classifier_95.onnx'
81
81
else :
82
82
url = 'https://models.silero.ai/vad_models/lang_classifier_95.jit'
83
83
model = Validator (url )
84
-
85
- with open (f'{ hub_dir } /snakers4_silero-vad_master/files/lang_dict_95.json' , 'r' ) as f :
84
+
85
+ model_dir = os .path .join (os .path .dirname (__file__ ), 'files' )
86
+ with open (os .path .join (model_dir , 'lang_dict_95.json' ), 'r' ) as f :
86
87
lang_dict = json .load (f )
87
88
88
- with open (f' { hub_dir } /snakers4_silero-vad_master/files/ lang_group_dict_95.json' , 'r' ) as f :
89
+ with open (os . path . join ( model_dir , ' lang_group_dict_95.json') , 'r' ) as f :
89
90
lang_group_dict = json .load (f )
90
91
91
92
utils = (get_language_and_group , read_audio )
0 commit comments