Update config.json

config.json

@@ -1,17 +1,41 @@
 {
   "name": "YOLOv8 Digit and Icon Detector",
   "framework": "onnx",
-  "id2label": {
-    "0": "digit",
-    "1": "icon"
-  },
-  "label2id": {
-    "digit": 0,
-    "icon": 1
-  },
-  "interpretation_script": "yolo",
-  "weights": "best1-2.onnx",
-  "image_size": [640, 640],
-  "framework": "onnx",
-  "inference_framework": "onnx"
+  "type": "detection",
+  "labels": [
+    {
+      "name": "digit",
+      "color": "#ff0000"
+    },
+    {
+      "name": "icon",
+      "color": "#00ff00"
+    }
+  ],
+  "model_attributes": [
+    {
+      "name": "confidence_threshold",
+      "default_value": 0.25,
+      "input_type": "number"
+    },
+    {
+      "name": "iou_threshold",
+      "default_value": 0.45,
+      "input_type": "number"
+    }
+  ],
+  "inputs": [
+    {
+      "name": "images",
+      "shape": [1, 3, 640, 640],
+      "layout": "NCHW"
+    }
+  ],
+  "outputs": [
+    {
+      "name": "output0",
+      "shape": [-1, 5]
+    }
+  ],
+  "interpretation_script": "yolo_v8_onnx_detector"
 }