def silero_lang_detector(**kwargs): """Silero Language Classifier 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') utils = (get_language, read_audio) return model, utils
def silero_vad(**kwargs): """Silero Voice Activity Detector 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/model.jit') utils = (get_speech_ts, save_audio, read_audio, state_generator, single_audio_stream, collect_chunks) return model, utils
def silero_number_detector(**kwargs): """Silero Number Detector 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') utils = (get_number_ts, save_audio, read_audio, collect_chunks, drop_chunks) return model, utils
def silero_vad(onnx=False): """Silero Voice Activity Detector 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() if onnx: model = OnnxWrapper( f'{hub_dir}/snakers4_silero-vad_master/files/silero_vad.onnx') else: model = init_jit_model( model_path= f'{hub_dir}/snakers4_silero-vad_master/files/silero_vad.jit') utils = (get_speech_timestamps, save_audio, read_audio, VADIterator, collect_chunks) return model, utils
def silero_lang_detector_116(**kwargs): """Silero Language Classifier (116 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') with open(f'{hub_dir}/snakers4_silero-vad_master/files/lang_dict_116.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: lang_group_dict = json.load(f) utils = (get_language_and_group, read_audio) return model, lang_dict, lang_group_dict, utils