docker-compose.yml

version: '3.8'

services:
  ncut_demo:
    image: registry.hf.space/huzey-ncut-pytorch:latest
    container_name: ncut_demo
    environment:
      - HF_ACCESS_TOKEN=${HF_ACCESS_TOKEN}
      - USE_HUGGINGFACE_ZEROGPU=false
    ports:
      - "7860:7860"
    deploy:
      resources:
        reservations:
          devices:
            - capabilities: [gpu]
    platform: linux/amd64
    runtime: nvidia
    shm_size: '64G'
    command: python app.py
    networks:
      - default