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

Polish hubconf file

Browse Source
This commit is contained in:
snakers41
2020-12-15 12:42:56 +00:00
parent 9df42530f8
commit ab55c0d738
1 changed files with 11 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
hubconf.py
View File

@@ -1,28 +1,25 @@
dependencies = ['torch', 'torchaudio', 'numpy'] dependencies = ['torch', 'torchaudio', 'numpy']
import torch import torch
from omegaconf import OmegaConf from omegaconf import OmegaConf
from utils import (init_jit_model, from utils import (init_jit_model,
get_speech_ts, get_speech_ts,
save_audio, save_audio,
read_audio, read_audio,
state_generator, state_generator,
single_audio_stream, single_audio_stream,
collect_speeches) collect_speeches)
def silero_stt(**kwargs): def silero_vad(**kwargs):
"""Silero Voice Activity and Number Detector Models """Silero Voice Activity Detector, Number Detector and Language Classifier
Returns a model and a set of utils Returns a model with a set of utils
Please see https://github.com/snakers4/silero-vad for usage examples Please see https://github.com/snakers4/silero-vad for usage examples
""" """
torch.hub.download_url_to_file('https://raw.githubusercontent.com/snakers4/silero-vad/master/files/model.jit', model = init_jit_model(model_path='files/model.jit')
'files/model.jit',
progress=False)
model = init_jit_model(model_url='files/model.jit')
utils = (get_speech_ts, utils = (get_speech_ts,
save_audio, save_audio,
read_audio, read_audio,
state_generator, state_generator,
single_audio_stream, single_audio_stream,
collect_speeches) collect_speeches)