subnet_bridge: copy winning miner repo into library

README.md

@@ -1,7 +1,7 @@
 ---
 tags:
 - element_type:detect
-- model:onnxruntime
+- model:ultralytics-yolo
 - subnet:winner
 - object:fire
 - object:smoke
@@ -10,18 +10,18 @@ tags:
 manako:
   source: winner_fetch
   manifest_element_name: manak0/Detect-fire
-  winner_repo_id: happydev93/firedetect1007
-  winner_revision: 89d9e46273bab6f91bfa5b9e2057e71f4dd52f4e
-  note: E=0.03070886 (map50=0.600000, size_mb=19.538334)
+  winner_repo_id: SuperBitDev/bev5
+  winner_revision: 42f54c2a500a1f78032404887c29a0d77e3942b3
+  note: E=0.03106038 (map50=0.600000, size_mb=19.317213)
 ---
 
 # Detect-fire-winner
 
 Published winning miner converted into a library element.
 
-- Source winner repo: `happydev93/firedetect1007`
-- Source revision: `89d9e46273bab6f91bfa5b9e2057e71f4dd52f4e`
+- Source winner repo: `SuperBitDev/bev5`
+- Source revision: `42f54c2a500a1f78032404887c29a0d77e3942b3`
 - Manifest element: `manak0/Detect-fire`
 - Element type: `Detect`
 - Objects: fire, smoke, fire extinguisher
-- Runtime model type: `onnxruntime`
+- Runtime model type: `ultralytics-yolo`

chute_config.yml

@@ -1,20 +1,20 @@
-Image:
-  from_base: parachutes/python:3.12
-  run_command:
-    - pip install --upgrade setuptools wheel
-    - pip install 'numpy>=1.23' 'onnxruntime-gpu[cuda,cudnn]>=1.16' 'opencv-python>=4.7' 'pillow>=9.5' 'huggingface_hub>=0.19.4' 'pydantic>=2.0' 'pyyaml>=6.0' 'aiohttp>=3.9'
-    - pip install torch torchvision
-
-NodeSelector:
-  gpu_count: 1
-  min_vram_gb_per_gpu: 16
-  include:
-    - pro_6000
-
-Chute:
-  timeout_seconds: 900
-  concurrency: 4
-  max_instances: 5
-  scaling_threshold: 0.5
-  shutdown_after_seconds: 288000
-  tee: true
+Image:
+  from_base: parachutes/python:3.12
+  run_command:
+    - pip install --upgrade setuptools wheel
+    - 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'
+    - pip install torch==2.8.0 torchvision==0.23.0 torchaudio==2.8.0 --index-url https://download.pytorch.org/whl/cu128
+
+NodeSelector:
+  gpu_count: 1
+  min_vram_gb_per_gpu: 16
+  include:
+    - pro_6000
+
+Chute:
+  timeout_seconds: 900
+  concurrency: 4
+  max_instances: 5
+  scaling_threshold: 0.5
+  shutdown_after_seconds: 288000
+  tee: true

main.py

@@ -28,7 +28,7 @@ Miner = _load_local_miner_class()
 
 
 CLASS_NAMES = ['fire', 'smoke', 'fire extinguisher']
-MODEL_TYPE = 'onnxruntime'
+MODEL_TYPE = 'ultralytics-yolo'
 
 
 def _to_dict(value: Any) -> dict[str, Any]:

pyproject.toml

@@ -5,13 +5,14 @@ requires-python = ">=3.9"
 
 dependencies = [
   "numpy>=1.23",
-  "onnxruntime[cuda,cudnn]>=1.16",
+  "onnxruntime>=1.16",
   "opencv-python>=4.7",
   "pillow>=9.5",
   "huggingface_hub>=0.19.4",
   "pydantic>=2.0",
   "pyyaml>=6.0",
   "aiohttp>=3.9",
-  "torch",
-  "torchvision",
+  "torch==2.8.0",
+  "torchvision==0.23.0",
+  "torchaudio==2.8.0",
 ]

weights.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:dbd2d9858bb969a4c3e83f95b2ff4bcbb34bb8add46b2bc6ac9fa72fb4966bf5
-size 19407313
+oid sha256:f7f53c40f57cc11d3923f3ca78984725bee6131511466c8163ef21c6fd625bfa
+size 19286315