Update inference_i2mv_sdxl.py

inference_i2mv_sdxl.py

@@ -114,7 +114,13 @@ def remove_bg(image: Image.Image, net, transform, device, mask: Image.Image = No
 def preprocess_image(image: Image.Image, height, width):
     image = np.array(image)
     # alpha = image[..., 3] > 0
-    alpha = image
+    # alpha = image
+
+    if image.mode in ("RGBA", "LA"):
+        alpha = image[..., 3]  # Extract the alpha channel
+    elif image.mode in ("RGB"):
+         # Create default alpha for non-alpha images
+         alpha = np.ones(image[..., 0].shape, dtype=np.uint8) * 255 # Create
     H, W = alpha.shape
     # get the bounding box of alpha
     y, x = np.where(alpha)