Update config.json

config.json

@@ -1,37 +1,34 @@
 {
-  "model_name": "Digit and Icon Detector",
+  "model_type": "object_detection",
   "framework": "onnx",
-  "task_type": "object-detection",
-  "architecture": "ONNX",
-  "labels": [
-    {
-      "id": 0,
-      "name": "digit",
-      "color": "#ff0000"
-    },
-    {
-      "id": 1,
-      "name": "icon",
-      "color": "#00ff00"
-    }
-  ],
+  "task": "object-detection",
+  "architecture": "YOLOv8",
+  "format": "ONNX",
+  "num_labels": 2,
+  "id2label": {
+    "0": "digit",
+    "1": "icon"
+  },
+  "label2id": {
+    "digit": 0,
+    "icon": 1
+  },
   "input_shape": [1, 3, 640, 640],
-  "input_format": "NCHW",
-  "preprocessing": {
-    "resize_mode": "letterbox",
-    "size": [640, 640],
-    "mean": [0, 0, 0],
-    "std": [255, 255, 255],
-    "to_rgb": true
+  "inference_framework": "onnxruntime",
+  "preprocessor": {
+    "type": "YOLOv8ImageProcessor",
+    "size": {
+      "height": 640,
+      "width": 640
+    },
+    "do_normalize": true,
+    "do_resize": true,
+    "resize_method": "letterbox"
   },
-  "postprocessing": {
+  "postprocessor": {
+    "type": "YOLOv8Postprocessor",
     "confidence_threshold": 0.25,
     "iou_threshold": 0.45,
-    "max_detections": 300
-  },
-  "metadata": {
-    "description": "ONNX model for detecting digits and icons",
-    "author": "Your Name",
-    "version": "1.0"
+    "max_detections_per_image": 300
   }
 }