fixing ensure img greyscaled

app.py

@@ -43,6 +43,7 @@ for example in dataset:
 # 3️⃣ Define preprocessing and inference function
 # ------------------------------------------------------------
 def preprocess_image(img: Image.Image):
+    img = img.convert("L")   # Ensure grayscale (1 channel): "L" = 8-bit grayscale
     img = img.resize(TARGET_SIZE, Image.BILINEAR)  # resize image to (target_height, target_width)
     img_array = img_to_array(img) / 255.0  # convert image to float32 NumPy array and normalize pixel values to [0, 1]
     img_array = np.expand_dims(img_array, axis=0)  # add batch dimension -> shape becomes (1, height, width, channels)

modelbuilder.py

@@ -61,7 +61,7 @@ class ModelBuilder:
             )
 
     def get_augmentation_pipe(self):
-        # Random-* layers are stochastic only when training=True
+        # Random/Augmentation layers are stochastic only when training=True
         # disabled during inference/evaluation
         return Sequential(
             [
@@ -78,12 +78,15 @@ class ModelBuilder:
 
         # Define input layer
         inputs = Input(shape=self.input_shape, name="inputs")
-        # Random-* layers are stochastic only when training=True
-
+        
+        # --- Random/Augmentation layers are stochastic only when training=True
         x_aug = self.get_augmentation_pipe()(
             inputs
-        )  # stochastic only when training=True
-        x = Rescaling(1.0 / 255)(x_aug)  # disabled during inference/evaluation
+        ) 
+        # ----- end augmentation -----
+        
+        # --- common preprocessing layer: rescaling to [0,1]
+        x = Rescaling(1.0 / 255)(x_aug)  
 
         # Model selector
         match self.model_type: