Update app.py

app.py

@@ -171,8 +171,16 @@ def remove_background_from_image(image: Image.Image) -> Image.Image:
     if result_image.mode != 'RGBA':
         result_image = result_image.convert('RGBA')
     
-    # Apply alpha mask
-    alpha_pil = Image.fromarray((alpha_mask * 255).astype(np.uint8), mode='L')
+    # Handle alpha mask dimensions and convert to PIL
+    # The alpha_mask might have extra dimensions, so squeeze and ensure 2D
+    alpha_mask_2d = np.squeeze(alpha_mask)
+    if alpha_mask_2d.ndim > 2:
+        # If still more than 2D, take the first channel
+        alpha_mask_2d = alpha_mask_2d[:, :, 0] if alpha_mask_2d.shape[-1] == 1 else alpha_mask_2d[:, :, 0]
+    
+    # Convert to uint8 and create PIL Image without deprecated mode parameter
+    alpha_array = (alpha_mask_2d * 255).astype(np.uint8)
+    alpha_pil = Image.fromarray(alpha_array, 'L')
     result_image.putalpha(alpha_pil)
     
     return result_image