Update config.json

config.json

@@ -1,44 +1,7 @@
 {
-  "model": {
-    "name": "ResNet34-based Image to Text Model",
-    "type": "Image to Text",
-    "input_shape": [224, 224, 3],
-    "output_shape": [1, "text"],
-    "base_model": "ResNet34",
-    "layers": [
-      {
-        "type": "Conv2D",
-        "filters": 64,
-        "kernel_size": [7, 7],
-        "activation": "relu",
-        "padding": "same",
-        "strides": [2, 2]
-      },
-      {
-        "type": "MaxPooling2D",
-        "pool_size": [3, 3],
-        "strides": [2, 2]
-      },
-      {
-        "type": "Residual Block",
-        "num_blocks": 34
-      },
-      {
-        "type": "GlobalAveragePooling2D"
-      },
-      {
-        "type": "Dense",
-        "units": 256,
-        "activation": "relu"
-      },
-      {
-        "type": "Dense",
-        "units": 1,
-        "activation": "sigmoid"
-      }
-    ],
-    "loss": "binary_crossentropy",
-    "optimizer": "adam",
-    "metrics": ["accuracy"]
-  }
-}
+  "model_type": "keras",
+  "framework": "tensorflow",
+  "image_size": [224, 224],
+  "num_labels": 28,
+  "model_name": "Retinal Disease Classification Model"
+}