mirror of
https://github.com/snakers4/silero-vad.git
synced 2026-02-04 17:39:22 +08:00
delete big files from repo
This commit is contained in:
12
utils_vad.py
12
utils_vad.py
@@ -5,6 +5,18 @@ import torch.nn.functional as F
|
||||
import warnings
|
||||
|
||||
languages = ['ru', 'en', 'de', 'es']
|
||||
onnx_url_dict = {
|
||||
'lang_classifier_95': 'https://models.silero.ai/vad_models/lang_classifier_95.onnx',
|
||||
'number_detector':'https://models.silero.ai/vad_models/number_detector.onnx'
|
||||
}
|
||||
|
||||
|
||||
def donwload_onnx_model(model_name):
|
||||
|
||||
if model_name not in ['lang_classifier_95', 'number_detector']:
|
||||
raise ValueError
|
||||
|
||||
torch.hub.download_url_to_file(onnx_url_dict[model_name], f'{model_name}.onnx')
|
||||
|
||||
|
||||
def validate(model,
|
||||
|
||||
Reference in New Issue
Block a user