onnxruntime-gpu deps

chute_config.yml

@@ -2,15 +2,11 @@ Image:
   from_base: parachutes/python:3.12
   run_command:
     - pip install --upgrade setuptools wheel
-    - pip install 'ultralytics>=8.4' 'torch>=2.4' 'torchvision' 'opencv-python-headless' 'pillow>=9.5' 'huggingface_hub>=0.19.4' 'pydantic>=2.0' 'numpy>=1.23'
-  set_workdir: /app
+    - pip install 'numpy>=1.23' 'onnxruntime-gpu>=1.16' 'opencv-python>=4.7' 'pillow>=9.5' 'huggingface_hub>=0.19.4' 'pydantic>=2.0' 'pyyaml>=6.0' 'aiohttp>=3.9'
 
 NodeSelector:
   gpu_count: 1
   min_vram_gb_per_gpu: 16
-  max_hourly_price_per_gpu: 2
-  include:
-    - pro_6000
 
 Chute:
   timeout_seconds: 900