services:
  skin-py-analysis:
    build: /data/www/pifu
    image: skin-py-gpu:1.0.0
    working_dir: /app
    ports:
      - "8000:8000"
    volumes:
      - /data/www/pifu:/app
      - /data/analysis:/data/analysis
    environment:
      TZ: Asia/Shanghai
      PYTHONUNBUFFERED: 1
      TASK_DB_PATH: /data/analysis/.worker/tasks.sqlite3
      MAX_CONCURRENT_TASKS: 6
      MAX_PENDING_TASKS: 100
      MODEL_PATH: /app/best.pt
      IMAGE_WORKERS: 4

      NVIDIA_VISIBLE_DEVICES: all
      NVIDIA_DRIVER_CAPABILITIES: compute,utility
    command: python api_server.py
    restart: unless-stopped
    gpus: all
    networks:
      - default
      - common-network

networks:
  default:
    external: true
    name: skin-network

  common-network:
    external: true