Upload 2 files

README.md

@@ -4,7 +4,7 @@ emoji: 🧩🖼️📦
 colorFrom: red
 colorTo: pink
 sdk: gradio
-sdk_version: 6.17.3
+sdk_version: 6.10.0
 python_version: "3.12"
 app_file: app.py
 pinned: true

utils.py

@@ -578,7 +578,12 @@ def create_mask_now(img, invert):
 
     time.sleep(0.5)
 
-    transparent_image = img["layers"][0]
+    layers = (img.get("layers") or []) if isinstance(img, dict) else []
+    if not layers:
+        background = img.get("background") if isinstance(img, dict) else None
+        return background, None
+
+    transparent_image = layers[0]
 
     # Extract the alpha channel
     alpha_channel = np.array(transparent_image)[:, :, 3]