Update config.json

config.json

@@ -1,34 +1,17 @@
 {
   "name": "YOLOv8 Digit and Icon Detector",
   "framework": "onnx",
-  "labels": [
-    {
-      "name": "digit",
-      "color": "#ff0000",
-      "attributes": []
-    },
-    {
-      "name": "icon",
-      "color": "#00ff00",
-      "attributes": []
-    }
-  ],
+  "id2label": {
+    "0": "digit",
+    "1": "icon"
+  },
+  "label2id": {
+    "digit": 0,
+    "icon": 1
+  },
   "interpretation_script": "yolo",
   "weights": "best1-2.onnx",
-  "class_to_idx": {"digit": 0, "icon": 1},
-  "model_parameters": {
-    "confidence_threshold": 0.5,
-    "nms_threshold": 0.4
-  },
-  "input_size": [640, 640],
-  "preprocessing": {
-    "resize": {
-      "width": 640,
-      "height": 640
-    },
-    "normalize": {
-      "mean": [0.485, 0.456, 0.406],
-      "std": [0.229, 0.224, 0.225]
-    }
-  }
+  "image_size": [640, 640],
+  "framework": "onnx",
+  "inference_framework": "onnx"
 }