Update config.json

config.json

@@ -1,10 +1,15 @@
 {
-    "framework": "pt",
-    "task": "object-detection",
+    "architectures": [
+        "YOLOv8"
+    ],
     "model_type": "yolov8",
-    "_name_or_path": "fbononibelloepoch/malaria-detection",
-    "architectures": ["YOLOv8"],
-    "model_format": "ultralytics",
-    "inference_framework": "ultralytics",
-    "classes": ["infected", "uninfected"]
+    "task": "object-detection",
+    "framework": "pt",
+    "num_classes": 2,
+    "classes": ["infected", "uninfected"],
+    "image_size": [640, 640],
+    "backbone": "yolov8n",
+    "confidence_threshold": 0.25,
+    "iou_threshold": 0.45,
+    "_name_or_path": "fbononibelloepoch/malaria-detection"
 }