Update utils.py

utils.py

@@ -34,11 +34,24 @@ 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(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
+def preprocess(image_or_path):
+    if isinstance(image_or_path, str):  # Check if the input is a string (image path)
+        img = Image.open(image_or_path)
+    else:  # Assume the input is a PIL Image object
+        img = image_or_path
+    
+    # Resize the image
+    resized_img = img.resize((256, 256), Image.LANCZOS)
+
+    # Convert the image to a numpy array
+    img_array = np.array(resized_img)
+
+    # Normalize if necessary
+    img_array = img_array / 255.0
+
+    # Add a batch dimension
+    img_array = np.expand_dims(img_array, axis=0)
+
     return img_array
 
 def model_arc():