Fix resize black gap

app.py

@@ -13,6 +13,7 @@ DEVICE = "cuda"
 
 MAX_SEED = np.iinfo(np.int32).max
 FIXED_DIMENSION = 900
+FIXED_DIMENSION = (FIXED_DIMENSION // 16) * 16
 
 SYSTEM_PROMPT = r"""This two-panel split-frame image showcases a furniture in as a product shot versus styled in a room.
 [LEFT] standalone product shot image the furniture on a white background.
@@ -46,15 +47,6 @@ else:
     pipe.to(DEVICE)
 
 
-def calculate_optimal_dimensions(image: Image.Image) -> tuple[int, int]:
-    width, height = image.size
-    # Ensure dimensions are multiples of 16
-    width = (width // 16) * 16
-    height = (height // 16) * 16
-
-    return int(width), int(height)
-
-
 @spaces.GPU
 def infer(
     furniture_image: Image.Image,
@@ -123,11 +115,6 @@ def infer(
     )
     mask.paste(_furniture_mask, (0, 0))
     mask.paste(_room_mask, (FIXED_DIMENSION, 0), _room_mask)
-
-    width, height = calculate_optimal_dimensions(image)
-    # Resize the image and mask to the optimal dimensions for the VAe
-    image = image.resize((width, height))
-    mask = mask.resize((width, height))
     # Invert the mask
     mask = ImageOps.invert(mask)
     # Blur the mask
@@ -142,8 +129,8 @@ def infer(
         prompt=[prompt] * batch_size,
         image=[image] * batch_size,
         mask_image=[mask] * batch_size,
-        height=height,
-        width=width,
+        height=FIXED_DIMENSION,
+        width=FIXED_DIMENSION * 2,
         guidance_scale=guidance_scale,
         num_inference_steps=num_inference_steps,
         generator=torch.Generator("cpu").manual_seed(seed),