Update config.json

config.json

@@ -1,8 +1,39 @@
 {
-  "architectures": ["ResNet"],
-  "num_labels": 2, 
-  "hidden_size": 512, 
-  "num_hidden_layers": 18, 
-  "output_attentions": false,
-  "output_hidden_states": false
+  "architectures": [
+    "ResNet50"
+  ],
+  "model_type": "tf_resnet",
+  "layers": {
+    "base_model": {
+      "weights": "imagenet",
+      "include_top": false
+    },
+    "classifier": {
+      "dense_1": {
+        "units": 64,
+        "activation": "relu",
+        "kernel_regularizer": {
+          "l2": 0.001
+        }
+      },
+      "dropout_1": {
+        "rate": 0.4
+      },
+      "dense_2": {
+        "units": 1,
+        "activation": "sigmoid"
+      }
+    }
+  },
+  "loss": "binary_crossentropy",
+  "optimizer": {
+    "type": "Adam",
+    "learning_rate": 1e-5
+  },
+  "metrics": ["accuracy", "recall", "precision"],
+  "description": "A custom ResNet50-based model fine-tuned for tuberculosis detection (binary classification).",
+  "preprocessing": {
+    "input_shape": [224, 224, 3],
+    "rescale": [0, 255]
+  }
 }