toxic-detector/docker-compose.yml

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