Update

app.py

@@ -27,6 +27,27 @@ def encode_image_as_base64(image: Image.Image) -> str:
     return base64.b64encode(buffered.getvalue()).decode("utf-8")
 
 
+def make_example(image_path: Path) -> EditorValue:
+    background_image = Image.open(image_path)
+    background_image = background_image.convert("RGB")
+    background = np.array(background_image)
+
+    mask = np.zeros_like(background)
+
+    layer = np.zeros((background.shape[0], background.shape[1], 4), dtype=np.uint8)
+    layer[:, :, 3] = mask
+
+    composite = np.zeros((background.shape[0], background.shape[1], 4), dtype=np.uint8)
+    composite[:, :, :3] = background
+    composite[:, :, 3] = np.where(mask == 255, 0, 255)  # noqa: PLR2004
+
+    return {
+        "background": background,
+        "layers": [layer],
+        "composite": composite,
+    }
+
+
 def predict(
     model_type: Literal["schnell", "dev", "pixart"],
     image_and_mask: EditorValue,
@@ -177,7 +198,7 @@ with gr.Blocks(css=css) as demo:
             )
             image_and_mask_examples = gr.Examples(
                 examples=[
-                    Image.open(path).convert("RGBA")
+                    make_example(path)
                     for path in Path("./examples/scenes").glob("*.png")
                 ],
                 label="Room examples",