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

feat: add traefik, simplify docker setup

Browse Source
This commit is contained in:
Felix Spöttel
2023-03-01 14:17:14 +01:00
parent 20f558584a
commit f27fe02958
16 changed files with 166 additions and 192 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
scripts/download_models.py
View File

@@ -1,8 +1,9 @@
import os
import sys
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)
model_name = sys.argv[1].strip()
_download(_MODELS[model_name], "/models/", False)
# if model_name != "large":
# _download(_MODELS[f"{model_name}.en"], "/models/", False)