Only generate 2

app.py

@@ -48,7 +48,7 @@ else:
     pipe.to(DEVICE)
 
 
-@spaces.GPU(duration=120)
+@spaces.GPU(duration=150)
 def infer(
     furniture_image: Image.Image,
     room_image: EditorValue,
@@ -135,7 +135,7 @@ def infer(
         width=FIXED_DIMENSION * 2,
         num_inference_steps=num_inference_steps,
         guidance_scale=guidance_scale,
-        num_images_per_prompt=4,
+        num_images_per_prompt=2,
         generator=torch.Generator("cpu").manual_seed(seed),
     )["images"]
 
@@ -171,7 +171,7 @@ with gr.Blocks(css=css) as demo:
                         type="pil",
                         sources=["upload"],
                         image_mode="RGB",
-                        height=300,
+                        height=400,
                     )
                     room_image = gr.ImageEditor(
                         label="Room Image - Draw mask for inpainting",
@@ -181,7 +181,7 @@ with gr.Blocks(css=css) as demo:
                         layers=False,
                         crop_size="1:1",
                         brush=gr.Brush(colors=["#FFFFFF"], color_mode="fixed"),
-                        height=300,
+                        height=400,
                     )
                 prompt = gr.Text(
                     label="Prompt",
@@ -197,7 +197,7 @@ with gr.Blocks(css=css) as demo:
                 format="png",
                 show_label=False,
                 columns=2,
-                height=600,
+                height=700,
             )
 
         with gr.Accordion("Advanced Settings", open=False):