Update README.md

README.md

@@ -98,10 +98,10 @@ def prepare_masked_image(
 
     # Normalize mask to [0,1] float if it’s in [0,255]
 
-    if mask_np.max() > 1:
-        mask_norm = (mask_np / 255.0).astype(np.float32)
-    else:
+    if mask_np.max() <= 1:
         mask_norm = mask_np.astype(np.float32)
+    else:
+        mask_norm = (mask_np / 255.0).astype(np.float32)
 
     # Expand mask to 3 channels if needed
     if mask_norm.ndim == 2: