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

fix typos

Browse Source
This commit is contained in:
Ziyuan Wang
2023-03-09 04:26:15 +00:00
parent 3d10c2d950
commit 9865b3cb93
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
examples/parallel_example.ipynb
View File

@@ -20,7 +20,7 @@
"from pprint import pprint\n", "from pprint import pprint\n",
"torch.set_num_threads(1)\n", "torch.set_num_threads(1)\n",
"# download wav files, make multiple copies\n", "# download wav files, make multiple copies\n",
"for idx in range(10):\n", "for idx in range(4):\n",
" torch.hub.download_url_to_file('https://models.silero.ai/vad_models/en.wav', f\"en_example{idx}.wav\")\n" " torch.hub.download_url_to_file('https://models.silero.ai/vad_models/en.wav', f\"en_example{idx}.wav\")\n"
] ]
}, },
@@ -41,7 +41,7 @@
"model, utils = torch.hub.load(repo_or_dir='snakers4/silero-vad',\n", "model, utils = torch.hub.load(repo_or_dir='snakers4/silero-vad',\n",
" model='silero_vad',\n", " model='silero_vad',\n",
" force_reload=True,\n", " force_reload=True,\n",
" onnx=USE_ONNX)\n", " onnx=False)\n",
"\n", "\n",
"(get_speech_timestamps,\n", "(get_speech_timestamps,\n",
" save_audio,\n", " save_audio,\n",
@@ -66,7 +66,7 @@
"source": [ "source": [
"def vad_process(audio_file: str):\n", "def vad_process(audio_file: str):\n",
" with torch.no_grad():\n", " with torch.no_grad():\n",
" wav = read_audio(audio_file, sample_rate=SAMPLING_RATE)\n", " wav = read_audio(audio_file, sampling_rate=SAMPLING_RATE)\n",
" return get_speech_timestamps(\n", " return get_speech_timestamps(\n",
" wav,\n", " wav,\n",
" model,\n", " model,\n",
@@ -98,7 +98,7 @@
"\n", "\n",
"futures = []\n", "futures = []\n",
"with ProcessPoolExecutor(max_workers=4) as ex:\n", "with ProcessPoolExecutor(max_workers=4) as ex:\n",
" for i in range(10):\n", " for i in range(4):\n",
" futures.append(ex.submit(vad_process, f\"en_example{idx}.wav\"))\n", " futures.append(ex.submit(vad_process, f\"en_example{idx}.wav\"))\n",
"\n", "\n",
"for finished in as_completed(futures):\n", "for finished in as_completed(futures):\n",