Update config.json

config.json

@@ -1,6 +1,15 @@
-
 {
   "pipeline_tag": "image-segmentation",
   "tags": ["yolov8", "segmentation"],
-  "model": "yolov8-seg"
+  "model": "yolov8-seg",
+  "class_set": [
+    {
+      "name": "Architrave",
+      "id": 0
+    },
+    {
+      "name": "Doojam",
+      "id": 1
+    }
+  ]
 }