Skip to content

Commit f9876dd

Browse files
committed
v3.1 path fix
1 parent 8ebaf13 commit f9876dd

File tree

1 file changed

+8
-7
lines changed

1 file changed

+8
-7
lines changed

hubconf.py

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
dependencies = ['torch', 'torchaudio']
22
import torch
3+
import os
34
import json
45
from 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

Comments
 (0)