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

delete big files from repo

Browse Source
This commit is contained in:
adamnsandle
2021-12-10 09:52:22 +00:00
parent 79fdb55f1c
commit f67e68efc3
7 changed files with 60 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
hubconf.py
View File

@@ -1,6 +1,7 @@
dependencies = ['torch', 'torchaudio']
import torch
import json
from utils_vad import (init_jit_model,
get_speech_timestamps,
get_number_ts,
@@ -10,7 +11,8 @@ from utils_vad import (init_jit_model,
read_audio,
VADIterator,
collect_chunks,
drop_chunks)
drop_chunks,
donwload_onnx_model)
def silero_vad(**kwargs):
@@ -34,13 +36,14 @@ def silero_number_detector(**kwargs):
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/number_detector.jit')
torch.hub.download_url_to_file('https://models.silero.ai/vad_models/number_detector.jit', 'number_detector.jit')
model = init_jit_model(model_path='number_detector.jit')
utils = (get_number_ts,
save_audio,
read_audio,
collect_chunks,
drop_chunks)
drop_chunks,
donwload_onnx_model)
return model, utils
@@ -50,10 +53,11 @@ def silero_lang_detector(**kwargs):
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/number_detector.jit')
torch.hub.download_url_to_file('https://models.silero.ai/vad_models/number_detector.jit', 'number_detector.jit')
model = init_jit_model(model_path='number_detector.jit')
utils = (get_language,
read_audio)
read_audio,
donwload_onnx_model)
return model, utils
@@ -65,7 +69,8 @@ def silero_lang_detector_95(**kwargs):
"""
hub_dir = torch.hub.get_dir()
model = init_jit_model(model_path=f'{hub_dir}/snakers4_silero-vad_master/files/lang_classifier_95.jit')
torch.hub.download_url_to_file('https://models.silero.ai/vad_models/lang_classifier_95.jit', 'lang_classifier_95.jit')
model = init_jit_model(model_path='lang_classifier_95.jit')
with open(f'{hub_dir}/snakers4_silero-vad_master/files/lang_dict_95.json', 'r') as f:
lang_dict = json.load(f)
@@ -73,6 +78,6 @@ def silero_lang_detector_95(**kwargs):
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)
utils = (get_language_and_group, read_audio, donwload_onnx_model)
return model, lang_dict, lang_group_dict, utils