38 lines
966 B
YAML
38 lines
966 B
YAML
services:
|
|
toxicity_api:
|
|
build: .
|
|
container_name: toxicity_assessment_api
|
|
ports:
|
|
- "8002:8002"
|
|
environment:
|
|
- MODEL_CHECKPOINT=cointegrated/rubert-tiny-toxicity
|
|
- USE_CUDA=False
|
|
- REDIS_URL=redis://redis:6379/0
|
|
- CELERY_BROKER_URL=redis://redis:6379/0
|
|
- 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.tasks worker --loglevel=info
|
|
environment:
|
|
- MODEL_CHECKPOINT=cointegrated/rubert-tiny-toxicity
|
|
- USE_CUDA=False
|
|
- REDIS_URL=redis://redis:6379/0
|
|
- CELERY_BROKER_URL=redis://redis:6379/0
|
|
- CELERY_RESULT_BACKEND=redis://redis:6379/0
|
|
depends_on:
|
|
- redis
|
|
restart: unless-stopped
|
|
|
|
redis:
|
|
image: redis:7-alpine
|
|
container_name: redis
|
|
ports:
|
|
- "6379:6379"
|
|
restart: unless-stopped
|