Update app.py

app.py

@@ -49,6 +49,13 @@ def inpaintGen(
     
     width, height = source_img.size
 
+    new_width = (width // 16) * 16
+    new_height = (height // 16) * 16
+        
+    # If the image size is not already divisible by 16, resize it
+    if width != new_width or height != new_height:
+        source_img = source_img.resize((new_width, new_height), Image.LANCZOS)
+
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator("cpu").manual_seed(seed)
@@ -57,8 +64,8 @@ def inpaintGen(
         prompt=inpaint_prompt,
         image=source_img,
         mask_image=binary_mask,
-        width=width,
-        height=height,
+        width=new_width,
+        height=new_height,
         num_inference_steps=num_steps,
         generator=generator,
         guidance_scale=guidance,