Update config.json

config.json

@@ -1,22 +1,23 @@
 {
-  "name": "aiweld-annotate-yolov8-seg",
-  "type": "detector",
-  "framework": "ultralytics",
-  "task": "segmentation",
-  "weights": {
-    "url": "best.pt"
-  },
+  "name": "YOLOv8 Weld Segmentation",
+  "framework": "pytorch",
+  "type": "segmentation",
+  "pipeline_tag": "image-segmentation",
   "labels": [
-    { "name": "Porosity" },
-    { "name": "Slag Inclusion" },
-    { "name": "Incomplete Penetration" },
-    { "name": "Incomplete Fusion" },
-    { "name": "Wormhole" },
-    { "name": "Crack" },
-    { "name": "IQI" },
-    { "name": "Norm" },
-    { "name": "Linear Porosity" },
-    { "name": "Artifact" },
-    { "name": "Weld" }
-  ]
+    { "name": "Porosity", "attributes": [] },
+    { "name": "Slag Inclusion", "attributes": [] },
+    { "name": "Incomplete Penetration", "attributes": [] },
+    { "name": "Incomplete Fusion", "attributes": [] },
+    { "name": "Wormhole", "attributes": [] },
+    { "name": "Crack", "attributes": [] },
+    { "name": "IQI", "attributes": [] },
+    { "name": "Norm", "attributes": [] },
+    { "name": "Linear Porosity", "attributes": [] },
+    { "name": "Artifact", "attributes": [] },
+    { "name": "Weld", "attributes": [] }
+  ],
+  "params": {
+    "threshold": 0.25
+  },
+  "api_url": "https://huggingface.co/aiweld/annotate"
 }