Image:
  from_base: parachutes/python:3.12
  set_workdir: /app

  run_command:
    - pip install --upgrade setuptools wheel

    - pip install --no-cache-dir torch==2.8.0 torchvision==0.23.0
    - pip install --no-cache-dir onnxruntime opencv-python-headless huggingface_hub pyyaml aiohttp pydantic numpy pillow

    - git clone https://github.com/BluxWarrior/yolo_onnx_v2

    - python /app/yolo_onnx_v2/install_ort_patch.py

    - python -c "import onnxruntime as ort, inspect; print('ORT', ort.__version__); print('Session', ort.InferenceSession); print('File', inspect.getfile(ort.InferenceSession))"

NodeSelector:
  gpu_count: 1
  min_vram_gb_per_gpu: 16
  max_hourly_price_per_gpu: 2
  include:
    - pro_6000

Chute:
  concurrency: 4
  max_instances: 5
  scaling_threshold: 0.5
  shutdown_after_seconds: 288000
  tee: true