Update utils.py

utils.py

@@ -10,7 +10,7 @@ def create_model():
     inputs = layers.Input(shape=(224, 224, 3), name="input_layer")
 
     x = baseModel(inputs)
-    x = layers.AveragePooling2D()(x)
+    x = layers.AveragePooling2D(pool_size=(2, 2))(x)
     x = layers.Flatten(name='Flatten')(x)
     x = layers.Dense(units=128, activation='relu')(x)
     x = layers.Dropout(rate=0.5)(x)