Rename model-index.json to config.json

config.json

@@ -0,0 +1,26 @@
+{
+  "architectures": ["MobileNet"],
+  "model_type": "mobilenet",
+  "framework": "tf",
+  "num_classes": 4,
+  "input_shape": [128, 128, 3],
+  "base_model": {
+    "name": "mobilenet",
+    "weights": "imagenet",
+    "include_top": false,
+    "trainable": false
+  },
+  "custom_layers": [
+    {"type": "GlobalAveragePooling2D"},
+    {"type": "Dense", "units": 128, "activation": "relu"},
+    {"type": "Dropout", "rate": 0.5},
+    {"type": "Dense", "units": 4, "activation": "softmax"}
+  ],
+  "optimizer": {
+    "name": "Adam",
+    "learning_rate": 0.001
+  },
+  "loss": "categorical_crossentropy",
+  "metrics": ["accuracy"],
+  "class_names": ["Fake", "Low", "Medium", "High"]
+}

model-index.json

@@ -1,5 +0,0 @@
-{
-  "name": "MobileNet_Fire",
-  "task": "image-classification",
-  "framework": "fastapi"
-}