Update config.json

config.json

@@ -3,12 +3,17 @@
   "image_size": [640, 640],
   "num_classes": 1,
   "segmentation": true,
-  "class_set": {
-    "class_labels": ["Grape"],
-    "num_classes": 1
-  },
   "pipeline_tag": "image-segmentation",
   "model_id": "Yasssh2123/Grape",
   "license": "other",
-  "description": "A YOLOv8 model for segmenting grape images."
+  "description": "A YOLOv8 model for segmenting grape images.",
+  "class_set": {
+    "classes": [
+      {
+        "id": 0,
+        "name": "Grape"
+      }
+    ],
+    "num_classes": 1
+  }
 }