Update config.json

config.json

@@ -1,19 +1,38 @@
 {
-  "model_type": "yolov8",
-  "image_size": [640, 640],
-  "num_classes": 1,
-  "segmentation": true,
-  "pipeline_tag": "image-segmentation",
-  "model_id": "Yasssh2123/Grape",
-  "license": "other",
-  "description": "A YOLOv8 model for segmenting grape images.",
-  "class_set": {
-    "classes": [
-      {
-        "id": 0,
-        "name": "Grape"
-      }
+    "input_size": 640,
+    "task": "instance-segmentation",
+    "ultralytics_version": "8.0.20",  // Adjust this based on your actual version
+    "model_type": "v8",
+    "pipeline_tag": "image-segmentation",
+    "tags": ["yolov8", "segmentation"],
+    "model": "yolov8-seg",
+    "class_set": [
+        {"name": "Grape", "id": 0}
     ],
-    "num_classes": 1
-  }
+    "labels": [
+        {"name": "Grape", "attributes": []}
+    ],
+    "id2label": {
+        "0": "Grape"
+    },
+    "hyperparameters": {
+        "epochs": 100,  // Adjust as needed
+        "imgsz": 640,
+        "batch_size": 16,  // Adjust based on your training setup
+        "learning_rate": 0.01,
+        "momentum": 0.9,
+        "weight_decay": 0.0005,
+        "optimizer": "SGD",  // Change if you used a different optimizer
+        "lr_scheduler": "StepLR",
+        "step_size": 10,
+        "gamma": 0.1
+    },
+    "architectures": ["YOLOv8"],
+    "model_name_or_path": "Yasssh2123/Grape",  // Your Hugging Face model repo
+    "use_cache": false,
+    "do_sample": false,
+    "early_stopping": true,
+    "fp16": false,
+    "num_beams": 1,
+    "use_legacy_prediction_head": false
 }