test

app.py

@@ -170,18 +170,12 @@ def process(
     width, height = calculate_image_dimensions_for_flux(original_resolution_wh=image.size)
     image = image.resize((width, height), Image.LANCZOS)
     mask = mask.resize((width, height), Image.LANCZOS)
-    if mask_inflation_slider:
-        mask_array = np.array(mask)
-        kernel = np.ones((mask_inflation_slider, mask_inflation_slider), np.uint8)
-        mask_array = cv2.dilate(mask_array, kernel, iterations=1)
-        mask = Image.fromarray(mask_array)
-    if mask_blur_slider:
-        mask = mask.filter(ImageFilter.GaussianBlur(radius=mask_blur_slider))
+    mask = process_mask(mask, mask_inflation=mask_inflation_slider, mask_blur=mask_blur_slider)
 
     if randomize_seed_checkbox:
         seed_slicer = random.randint(0, MAX_SEED)
     generator = torch.Generator().manual_seed(seed_slicer)
-    result = pipe(
+    result = PIPE(
         prompt=inpainting_prompt_text,
         image=image,
         mask_image=mask,