Update utils.py

utils.py

@@ -34,12 +34,12 @@ base_model = tf.keras.applications.ResNet50V2(include_top=False, input_shape=inp
 #Making the layers of the model trainable
 base_model.trainable = True
 
-def preprocess(image):
-    image = np.array(image.resize((256, 256), Image.LANCZOS))
-    image = np.array(image, dtype='uint8')
-    image = np.array(image)/255.0
-
-    return image
+def preprocess(img_path):
+    img = Image.open(img_path).resize((256, 256), Image.LANCZOS)
+    img_array = np.array(img)
+    img_array = img_array / 255.0  # Normalize if necessary
+    img_array = np.expand_dims(img_array, axis=0)  # Add batch dimension
+    return img_array
 
 def model_arc():
     model = tf.keras.Sequential([