deepgeek/silero-vad
2
0
Fork 0
mirror of https://github.com/snakers4/silero-vad.git synced 2026-02-04 09:29:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

add pip examples to collab

Browse Source
This commit is contained in:
adamnsandle
2024-07-09 10:20:50 +00:00
parent dfdc9a484e
commit d9a6941852
1 changed files with 20 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
silero-vad.ipynb
View File

@@ -43,20 +43,30 @@
}, },
"outputs": [], "outputs": [],
"source": [ "source": [
"USE_PIP = True # download model using pip package or torch.hub\n",
"USE_ONNX = False # change this to True if you want to test onnx model\n", "USE_ONNX = False # change this to True if you want to test onnx model\n",
"if USE_ONNX:\n", "if USE_ONNX:\n",
" !pip install -q onnxruntime\n", " !pip install -q onnxruntime\n",
"if USE_PIP:\n",
" !pip install -q silero-vad\n",
" from silero_vad import (load_silero_vad,\n",
" read_audio,\n",
" get_speech_timestamps,\n",
" save_audio,\n",
" VADIterator,\n",
" collect_chunks)\n",
" model = load_silero_vad(onnx=USE_ONNX)\n",
"else:\n",
" model, utils = torch.hub.load(repo_or_dir='snakers4/silero-vad',\n",
" model='silero_vad',\n",
" force_reload=True,\n",
" onnx=USE_ONNX)\n",
"\n", "\n",
"model, utils = torch.hub.load(repo_or_dir='snakers4/silero-vad',\n", " (get_speech_timestamps,\n",
" model='silero_vad',\n", " save_audio,\n",
" force_reload=True,\n", " read_audio,\n",
" onnx=USE_ONNX)\n", " VADIterator,\n",
"\n", " collect_chunks) = utils"
"(get_speech_timestamps,\n",
" save_audio,\n",
" read_audio,\n",
" VADIterator,\n",
" collect_chunks) = utils"
] ]
}, },
{ {