mirror of
https://github.com/bellingcat/whisperbox-transcribe.git
synced 2026-06-07 19:18:35 +03:00
fix: configure docker build to use WHISPER_MODEL env
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user