Update config.json

config.json

@@ -1,10 +1,7 @@
 {
   "name": "YOLOv8 Digit and Icon Detector",
   "framework": "onnx",
-  "task_type": "detection",
-  "architecture": "yolov8",
-  "model_type": "yolov8",
-  "weights": "best1-2.onnx",
+  "type": "detection",
   "labels": [
     {
       "name": "digit",
@@ -15,19 +12,30 @@
       "color": "#00ff00"
     }
   ],
-  "id2label": {
-    "0": "digit",
-    "1": "icon"
-  },
-  "label2id": {
-    "digit": 0,
-    "icon": 1
-  },
-  "interpretation_script": "yolo",
-  "config": {
-    "input_size": [640, 640],
-    "confidence_threshold": 0.25,
-    "iou_threshold": 0.45,
-    "max_detections": 300
-  }
+  "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"
 }