Update utils.py

utils.py

@@ -37,8 +37,11 @@ base_model.trainable = True
 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
+    elif isinstance(image_or_path, np.ndarray):  # Check if the input is a numpy array
+        # Convert the numpy array to a PIL Image
+        img = Image.fromarray((image_or_path * 255).astype(np.uint8))
+    else:
+        raise ValueError("Unsupported input type. Expected string path or numpy array.")
     
     # Print image dimensions for debugging
     print(f"Original image dimensions: {img.size}")