deepgeek/silero-vad
2
0
Fork 0
mirror of https://github.com/snakers4/silero-vad.git synced 2026-02-05 18:09:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

95 classifier

Browse Source
This commit is contained in:
adamnsandle
2021-07-20 09:14:50 +00:00
parent ce70e18e67
commit b62c8af8a8
10 changed files with 20 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
hubconf.py
View File

@@ -134,19 +134,19 @@ def silero_lang_detector(**kwargs):
return model, utils
def silero_lang_detector_116(**kwargs):
"""Silero Language Classifier (116 languages)
def silero_lang_detector_95(**kwargs):
"""Silero Language Classifier (95 languages)
Returns a model with a set of utils
Please see https://github.com/snakers4/silero-vad for usage examples
"""
hub_dir = torch.hub.get_dir()
model = init_jit_model(model_path=f'{hub_dir}/snakers4_silero-vad_master/files/lang_classifier_116.jit')
model = init_jit_model(model_path=f'{hub_dir}/snakers4_silero-vad_master/files/lang_classifier_95.jit')
with open(f'{hub_dir}/snakers4_silero-vad_master/files/lang_dict_116.json', 'r') as f:
with open(f'{hub_dir}/snakers4_silero-vad_master/files/lang_dict_95.json', 'r') as f:
lang_dict = json.load(f)
with open(f'{hub_dir}/snakers4_silero-vad_master/files/lang_group_dict_116.json', 'r') as f:
with open(f'{hub_dir}/snakers4_silero-vad_master/files/lang_group_dict_95.json', 'r') as f:
lang_group_dict = json.load(f)
utils = (get_language_and_group, read_audio)