diff --git a/docker-compose.yml b/docker-compose.yml index 0cf959d..49b3b63 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -12,12 +12,13 @@ services: - CELERY_RESULT_BACKEND=redis://redis:6379/0 depends_on: - redis + - celery_worker restart: unless-stopped celery_worker: build: . container_name: celery_worker - command: celery -A app.worker worker --loglevel=info + command: celery -A app.tasks worker --loglevel=info environment: - MODEL_CHECKPOINT=cointegrated/rubert-tiny-toxicity - USE_CUDA=False