fix resize

app.py

@@ -21,6 +21,24 @@ def sanitize_prompt(prompt):
   sanitized_prompt = allowed_chars.sub("", prompt)
   return sanitized_prompt
 
+def convert_to_fit_size(original_width_and_height, maximum_size = 2048):
+    width, height =original_width_and_height
+    if width <= maximum_size and height <= maximum_size:
+        return width,height
+    
+    if width > height:
+        scaling_factor = maximum_size / width
+    else:
+        scaling_factor = maximum_size / height
+
+    new_width = int(width * scaling_factor)
+    new_height = int(height * scaling_factor)
+    return new_width, new_height
+
+def adjust_to_multiple_of_32(width: int, height: int):
+    width = width - (width % 32)
+    height = height - (height % 32)
+    return width, height
 
 
 
@@ -40,8 +58,15 @@ def process_images(image,prompt="a girl",strength=0.75,seed=0,inference_step=4,p
         generators = []
         generator = torch.Generator(device).manual_seed(seed)
         generators.append(generator)
+        width,height = convert_to_fit_size(image.size)
+        #print(f"fit {width}x{height}")
+        width,height = adjust_to_multiple_of_32(width,height)
+        #print(f"multiple {width}x{height}")
+        image = image.resize((width, height), Image.LANCZOS)
+        #mask_image = mask_image.resize((width, height), Image.NEAREST)
+
         # more parameter see https://huggingface.co/docs/diffusers/api/pipelines/flux#diffusers.FluxInpaintPipeline
-        print(prompt)
+        #print(prompt)
         output = pipe(prompt=prompt, image=image,generator=generator,strength=strength
                     ,guidance_scale=0,num_inference_steps=num_inference_steps,max_sequence_length=256)