Update config.json

config.json

@@ -1,11 +1,13 @@
 {
     "input_size": 640,
     "task": "instance-segmentation",
-    "model_type": "YOLOv8",
-    "pipeline_tag": "image-segmentation",
-    "tags": ["yolov8", "segmentation"],
+    "ultralytics_version": "8.0.20",
+    "model_type": "yolov8",
+    "pipeline_tag": "mask-generation",
+    "tags": [
+        "image-segmentation"
+    ],
     "model": "yolov8-seg",
-    "model_name_or_path": "Yasssh2123/Grape",
     "class_set": [
         {
             "name": "Grape",
@@ -20,5 +22,32 @@
     ],
     "id2label": {
         "0": "Grape"
-    }
+    },
+    "hyperparameters": {
+        "epochs": 100,
+        "imgsz": 640,
+        "batch_size": 16,
+        "learning_rate": 0.01,
+        "momentum": 0.9,
+        "weight_decay": 0.0005,
+        "optimizer": "SGD",
+        "lr_scheduler": "StepLR",
+        "step_size": 10,
+        "gamma": 0.1
+    },
+    "architectures": [
+        "YOLOv8"
+    ],
+    "model_name_or_path": "Yasssh2123/Grape",
+    "use_cache": false,
+    "do_sample": false,
+    "early_stopping": true,
+    "fp16": false,
+    "num_beams": 1,
+    "use_legacy_prediction_head": false,
+    "license": "agpl-3.0",
+    "base_model": [
+        "Ultralytics/YOLOv8",
+        "raidavid/yolov8n-seg"
+    ]
 }