Update config.json

config.json

@@ -1,4 +1,39 @@
 {
-  "model_type": "custom",
-  "pipeline_tag": "custom"
-}
+  "model_name": "CustomUNet",
+  "description": "A custom U-Net model for semantic segmentation using TensorFlow/Keras.",
+  "input_shape": [512, 512, 3],  // Input shape of the images
+  "num_classes": 11,  // Number of output classes
+  "architecture": {
+    "encoder": {
+      "type": "custom",
+      "blocks": [
+        {"filters": 64},
+        {"filters": 128},
+        {"filters": 256},
+        {"filters": 512}
+      ]
+    },
+    "bottleneck": {
+      "filters": 1024
+    },
+    "decoder": {
+      "type": "custom",
+      "blocks": [
+        {"filters": 512},
+        {"filters": 256},
+        {"filters": 128},
+        {"filters": 64}
+      ]
+    }
+  },
+  "loss_function": "categorical_crossentropy",
+  "optimizer": {
+    "type": "Adam",
+    "learning_rate": 0.001
+  },
+  "training": {
+    "batch_size": 16,
+    "epochs": 50,
+    "validation_split": 0.2
+  }
+}