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

fix: configure docker build to use WHISPER_MODEL env

Browse Source
This commit is contained in:
Felix Spöttel
2023-06-28 11:07:23 +02:00
parent c41d3c6dd1
commit 136852afda
2 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
docker-compose.base.yml
View File

@@ -27,14 +27,13 @@ services:
context: . context: .
dockerfile: worker.Dockerfile dockerfile: worker.Dockerfile
args: args:
# FIXME WHISPER_MODEL: ${WHISPER_MODEL}
WHISPER_MODEL: small
networks: networks:
- app - app
depends_on: depends_on:
- redis - redis
healthcheck: healthcheck:
test: ["CMD-SHELL", "celery -b redis://redis:6379/0 inspect ping -d celery@$$HOSTNAME"] test: ["CMD-SHELL", "celery -b ${BROKER_URL} inspect ping -d celery@$$HOSTNAME"]
interval: 5s interval: 5s
timeout: 5s timeout: 5s
retries: 5 retries: 5

2
scripts/download_models.py
View File

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