ijosdj

app.py

@@ -50,7 +50,7 @@ else:
 
 @spaces.GPU
 def infer(
-    furniture_image: EditorValue,
+    furniture_image: Image.Image,
     room_image: EditorValue,
     prompt: str = "",
     seed: int = 42,
@@ -84,7 +84,7 @@ def infer(
     )
 
     furniture_image = ImageOps.fit(
-        furniture_image["background"],
+        furniture_image,
         (FIXED_DIMENSION, FIXED_DIMENSION),
         method=Image.Resampling.LANCZOS,
         centering=(0.5, 0.5),
@@ -121,7 +121,7 @@ def infer(
     # Blur the mask
     mask = mask.filter(ImageFilter.GaussianBlur(radius=10))
     # Convert to 3 channel
-    mask = mask.convert("RGB")
+    mask = mask.convert("L")
 
     if randomize_seed:
         seed = secrets.randbelow(MAX_SEED)
@@ -168,16 +168,11 @@ with gr.Blocks(css=css) as demo:
         with gr.Row():
             with gr.Column():
                 with gr.Column():
-                    furniture_image = gr.ImageEditor(
+                    furniture_image = gr.Image(
                         label="Furniture Image",
                         type="pil",
                         sources=["upload"],
                         image_mode="RGB",
-                        crop_size="1:1",
-                        transforms=[],
-                        eraser=False,
-                        brush=False,
-                        layers=False,
                         height=300,
                     )
                     room_image = gr.ImageEditor(