Update utils.py

utils.py

@@ -34,8 +34,14 @@ 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)
+def preprocess(img_path_or_img):
+    if isinstance(img_path_or_img, str):  # Check if input is a path
+        img = Image.open(img_path_or_img)
+    elif isinstance(img_path_or_img, Image.Image):  # Check if input is already an Image object
+        img = img_path_or_img
+    else:
+        raise ValueError("Input must be a string path or a PIL Image object.")
+    
     img = img.resize((256, 256))
     img_array = np.array(img)
     img_array = np.expand_dims(img_array, axis=0)