Update config.json

config.json

@@ -1,23 +1,34 @@
 {
-  "architectures": ["YOLOv8"],
-  "model_type": "yolov8",
-  "task_specific_params": {
-    "object-detection": {
-      "confidence_threshold": 0.25,
-      "iou_threshold": 0.45
+  "name": "YOLOv8 Digit and Icon Detector",
+  "framework": "onnx",
+  "labels": [
+    {
+      "name": "digit",
+      "color": "#ff0000",
+      "attributes": []
+    },
+    {
+      "name": "icon",
+      "color": "#00ff00",
+      "attributes": []
     }
+  ],
+  "interpretation_script": "yolo",
+  "weights": "digit.onnx",
+  "class_to_idx": {"digit": 0, "icon": 1},
+  "model_parameters": {
+    "confidence_threshold": 0.5,
+    "nms_threshold": 0.4
   },
-  "id2label": {
-    "0": "digit",
-    "1": "icon"
-  },
-  "label2id": {
-    "digit": 0,
-    "icon": 1
-  },
-  "image_size": [640, 640],
-  "framework": "onnx",
-  "model_file": "best1-2.onnx",
-  "export_format": "onnx",
-  "social_visibility": "increase"
-}
+  "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]
+    }
+  }
+}