mirror of
https://github.com/snakers4/silero-vad.git
synced 2026-02-04 17:39:22 +08:00
29 lines
987 B
Python
29 lines
987 B
Python
dependencies = ['torch', 'omegaconf', 'torchaudio']
|
|
import torch
|
|
from omegaconf import OmegaConf
|
|
from utils import (init_jit_model,
|
|
read_audio,
|
|
read_batch,
|
|
split_into_batches,
|
|
prepare_model_input)
|
|
|
|
|
|
def silero_stt(**kwargs):
|
|
"""Silero Voice Activity and Number Detector Models
|
|
Returns a model and a set of utils
|
|
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/models.yml',
|
|
'silero_vad_models.yml',
|
|
progress=False)
|
|
models = OmegaConf.load('silero_vad_models.yml')
|
|
|
|
model = init_jit_model(model_url=models.latest.jit,
|
|
**kwargs)
|
|
utils = (read_batch,
|
|
split_into_batches,
|
|
read_audio,
|
|
prepare_model_input)
|
|
|
|
return model, utils
|