Update

app.py

@@ -28,7 +28,7 @@ def encode_image_as_base64(image: Image.Image) -> str:
 
 
 def predict(
-    model_type: Literal["schnell", "dev"],
+    model_type: Literal["schnell", "dev", "pixart"],
     image_and_mask: EditorValue,
     furniture_reference: Image.Image | None,
     prompt: str = "",
@@ -47,6 +47,10 @@ def predict(
         gr.Info("Please upload a furniture reference image")
         return None
 
+    if model_type == "pixart":
+        gr.Info("PixArt is not supported yet")
+        return None
+
     image_np = image_and_mask["background"]
     image_np = cast(np.ndarray, image_np)
 
@@ -172,7 +176,14 @@ with gr.Blocks(css=css) as demo:
                 transforms=[],
             )
             image_and_mask_examples = gr.Examples(
-                examples=list(Path("./examples/scenes").glob("*.png")),
+                examples=[
+                    {
+                        "background": path,
+                        "layers": None,
+                        "composite": None,
+                    }
+                    for path in Path("./examples/scenes").glob("*.png")
+                ],
                 label="Room examples",
                 examples_per_page=6,
                 inputs=[image_and_mask],
@@ -221,7 +232,7 @@ with gr.Blocks(css=css) as demo:
                 preview=True,
             )
             model_type = gr.Radio(
-                choices=["schnell", "dev"],
+                choices=["schnell", "dev", "pixart"],
                 value="schnell",
                 label="Model Type",
             )