version: '3.8'

services:
  vit-auditing-toolkit:
    build: .
    container_name: vit-auditing-toolkit
    ports:
      - "7860:7860"
    environment:
      - GRADIO_SERVER_NAME=0.0.0.0
      - GRADIO_SERVER_PORT=7860
    volumes:
      - ./models:/app/models  # Cache downloaded models
      - ./examples:/app/examples  # Mount example images
    restart: unless-stopped
    healthcheck:
      test: ["CMD", "curl", "-f", "http://localhost:7860/"]
      interval: 30s
      timeout: 10s
      retries: 3
      start_period: 60s