Update inference_i2mv_sdxl.py

inference_i2mv_sdxl.py

@@ -112,15 +112,17 @@ 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
 
     if image.mode in ("RGBA", "LA"):
+        image = np.array(image)
         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
+        image = np.array(image)
+        # 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)