version: '3.8'

services:
  vr180-matting:
    build: .
    runtime: nvidia
    environment:
      - NVIDIA_VISIBLE_DEVICES=all
      - CUDA_VISIBLE_DEVICES=0
    volumes:
      - ./data:/app/data          # Mount data directory
      - ./output:/app/output      # Mount output directory
      - ./models:/app/models      # Mount models directory
    working_dir: /app
    stdin_open: true
    tty: true
    command: /bin/bash