Force to crop

app.py

@@ -50,7 +50,7 @@ else:
 
 @spaces.GPU
 def infer(
-    furniture_image: Image.Image,
+    furniture_image: EditorValue,
     room_image: EditorValue,
     prompt: str = "",
     seed: int = 42,
@@ -84,7 +84,7 @@ def infer(
     )
 
     furniture_image = ImageOps.fit(
-        furniture_image,
+        furniture_image["background"],
         (FIXED_DIMENSION, FIXED_DIMENSION),
         method=Image.Resampling.LANCZOS,
         centering=(0.5, 0.5),
@@ -131,7 +131,7 @@ def infer(
     results_images = pipe(
         prompt=[prompt] * batch_size,
         image=[image] * batch_size,
-        mask_image=mask,
+        mask_image=[mask] * batch_size,
         height=FIXED_DIMENSION,
         width=FIXED_DIMENSION * 2,
         guidance_scale=guidance_scale,
@@ -168,11 +168,16 @@ with gr.Blocks(css=css) as demo:
         with gr.Row():
             with gr.Column():
                 with gr.Column():
-                    furniture_image = gr.Image(
+                    furniture_image = gr.ImageEditor(
                         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(
@@ -181,6 +186,7 @@ with gr.Blocks(css=css) as demo:
                         sources=["upload"],
                         image_mode="RGBA",
                         layers=False,
+                        crop_size="1:1",
                         brush=gr.Brush(colors=["#FFFFFF"], color_mode="fixed"),
                         height=300,
                     )