fix: configure docker build to use WHISPER_MODEL env

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

View File

@@ -27,14 +27,13 @@ services:
context: .
dockerfile: worker.Dockerfile
args:
# FIXME
WHISPER_MODEL: small
WHISPER_MODEL: ${WHISPER_MODEL}
networks:
- app
depends_on:
- redis
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
timeout: 5s
retries: 5

View File

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