Update utils.py

utils.py

@@ -34,10 +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 = image.astype('float32') / 255.0
-    return image
+def preprocess(img_path):
+    img = Image.open(img_path)
+    img = img.resize((256, 256))
+    img_array = np.array(img)
+    img_array = np.expand_dims(img_array, axis=0)
+    return img_array
 
 def model_arc():
     model = tf.keras.Sequential([