Update config.json

config.json

@@ -1,41 +1,21 @@
 {
-  "name": "YOLOv8 Digit and Icon Detector",
-  "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"
+  "architectures": ["YOLOv8"],
+  "model_type": "yolov8",
+  "task_specific_params": {
+    "object-detection": {
+      "confidence_threshold": 0.25,
+      "iou_threshold": 0.45
     }
-  ],
-  "outputs": [
-    {
-      "name": "output0",
-      "shape": [-1, 5]
-    }
-  ],
-  "interpretation_script": "yolo_v8_onnx_detector"
+  },
+  "id2label": {
+    "0": "digit",
+    "1": "icon"
+  },
+  "label2id": {
+    "digit": 0,
+    "icon": 1
+  },
+  "image_size": [640, 640],
+  "framework": "onnx",
+  "inference_framework": "onnx"
 }