This commit is contained in:
adamnsandle
2020-12-15 14:05:12 +00:00
parent a21e74886f
commit 6572cad0fa
2 changed files with 162 additions and 149 deletions

View File

@@ -14,7 +14,8 @@ def silero_vad(**kwargs):
Returns a model with a set of utils
Please see https://github.com/snakers4/silero-vad for usage examples
"""
model = init_jit_model(model_path='files/model.jit')
hub_dir = torch.hub.get_dir()
model = init_jit_model(model_path=f'{hub_dir}/files/model.jit')
utils = (get_speech_ts,
save_audio,
read_audio,