Update app.py

app.py

@@ -34,10 +34,10 @@ validation_generator = test_datagen.flow_from_directory(
 base_model = MobileNetV2(weights='imagenet', include_top=False, input_shape=(150, 150, 3))
 base_model.trainable = False  # Freeze the base model
 
-model = models.Sequential([
+model = tf.keras.models.Sequential([
     base_model,
     layers.GlobalAveragePooling2D(),
-    layers.Dense(128, activation='relu'),
+    layers.Dense(1024, activation='relu'),
     layers.Dense(len(labels), activation='softmax')
 ])