Update app.py

app.py

@@ -193,14 +193,15 @@ test_images = test_generator.flow_from_dataframe(
 
 # Data Augmentation Step
 augment = tf.keras.Sequential([
-    layers.experimental.preprocessing.Resizing(224, 224),
-    layers.experimental.preprocessing.Rescaling(1./255),
-    layers.experimental.preprocessing.RandomFlip("horizontal"),
-    layers.experimental.preprocessing.RandomRotation(0.1),
-    layers.experimental.preprocessing.RandomZoom(0.1),
-    layers.experimental.preprocessing.RandomContrast(0.1),
+    tf.keras.layers.Resizing(224, 224),
+    tf.keras.layers.Rescaling(1./255),
+    tf.keras.layers.RandomFlip("horizontal"),
+    tf.keras.layers.RandomRotation(0.1),
+    tf.keras.layers.RandomZoom(0.1),
+    tf.keras.layers.RandomContrast(0.1),
 ])
 
+
 # Load the pretrained model
 pretrained_model = tf.keras.applications.efficientnet_v2.EfficientNetV2L(
     input_shape=(224, 224, 3),