bellingcat/whisperbox-transcribe
1
0
Fork 0
mirror of https://github.com/bellingcat/whisperbox-transcribe.git synced 2026-06-13 05:58:35 +03:00
Code Issues Packages Projects Releases Wiki Activity

feat: allow multiple models to be installed in docker container

Browse Source
This commit is contained in:
Felix Spöttel
2023-02-28 22:05:09 +01:00
parent 099a0eb1d7
commit 20f558584a
6 changed files with 18 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
scripts/download_models.py Normal file
View File

@@ -0,0 +1,8 @@
import os
from whisper import _download, _MODELS # type: ignore
if __name__ == "__main__":
for name in os.environ["WHISPER_MODELS"].split(","):
_download(_MODELS[name], "/models/", False)
if name != "large":
_download(_MODELS[f"{name}.en"], "/models/", False)