11dependencies = ['torch' , 'torchaudio' ]
22import torch
3+ import os
34import json
45from utils_vad import (init_jit_model ,
56 get_speech_timestamps ,
@@ -20,11 +21,11 @@ def silero_vad(onnx=False):
2021 Returns a model with a set of utils
2122 Please see https://github.com/snakers4/silero-vad for usage examples
2223 """
23- hub_dir = torch . hub . get_dir ( )
24+ model_dir = os . path . join ( os . path . dirname ( __file__ ), 'files' )
2425 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') )
2627 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') )
2829 utils = (get_speech_timestamps ,
2930 save_audio ,
3031 read_audio ,
@@ -75,17 +76,17 @@ def silero_lang_detector_95(onnx=False):
7576 Please see https://github.com/snakers4/silero-vad for usage examples
7677 """
7778
78- hub_dir = torch .hub .get_dir ()
7979 if onnx :
8080 url = 'https://models.silero.ai/vad_models/lang_classifier_95.onnx'
8181 else :
8282 url = 'https://models.silero.ai/vad_models/lang_classifier_95.jit'
8383 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 :
8687 lang_dict = json .load (f )
8788
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 :
8990 lang_group_dict = json .load (f )
9091
9192 utils = (get_language_and_group , read_audio )
0 commit comments