Update config.json

config.json

@@ -1,25 +1,25 @@
 {
   "model_type": "Sequential",
+  "base_model": {
+    "name": "MobileNetV2",
+    "input_shape": [150, 150, 3],
+    "include_top": false,
+    "weights": "imagenet",
+    "trainable": false
+  },
   "layers": [
     {
-      "type": "Conv2D",
-      "filters": 32,
-      "kernel_size": [3, 3],
-      "activation": "relu",
-      "input_shape": [64, 64, 3]
-    },
-    {
-      "type": "MaxPooling2D",
-      "pool_size": [2, 2]
-    },
-    {
-      "type": "Flatten"
+      "type": "GlobalAveragePooling2D"
     },
     {
       "type": "Dense",
       "units": 128,
       "activation": "relu"
     },
+    {
+      "type": "Dropout",
+      "rate": 0.3
+    },
     {
       "type": "Dense",
       "units": 10,