bellingcat/whisperbox-transcribe
1
0
Fork 0
mirror of https://github.com/bellingcat/whisperbox-transcribe.git synced 2026-06-12 05:28:34 +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_model.py
View File

@@ -1,8 +0,0 @@
import sys
from whisper import _download, _MODELS # type: ignore
if __name__ == "__main__":
model_name = sys.argv[1]
_download(_MODELS[model_name], "/models/", False)
if model_name != "large":
_download(_MODELS[f"{model_name}.en"], "/models/", False)

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)