cap image size, ensure result dimension is divisible by 8

app.py

@@ -1,8 +1,9 @@
 from typing import Tuple
 
-import torch
-import spaces
 import gradio as gr
+import spaces
+import torch
+from PIL import Image
 from diffusers import FluxInpaintPipeline
 
 MARKDOWN = """
@@ -49,22 +50,24 @@ def process(input_image_editor, input_text, progress=gr.Progress(track_tqdm=True
         return None
 
     image = input_image_editor['background']
-    mask_image = input_image_editor['layers'][0]
+    mask = input_image_editor['layers'][0]
 
     if not image:
         gr.Info("Please upload an image.")
         return None
 
-    if not mask_image:
+    if not mask:
         gr.Info("Please draw a mask on the image.")
         return None
 
     width, height = resize_image_dimensions(original_resolution_wh=image.size)
+    resized_image = image.resize((width, height), Image.LANCZOS)
+    resized_mask = mask.resize((width, height), Image.NEAREST)
 
     return pipe(
         prompt=input_text,
-        image=image,
-        mask_image=mask_image,
+        image=resized_image,
+        mask_image=resized_mask,
         width=width,
         height=height,
         strength=0.7,