Update app.py

app.py

@@ -147,23 +147,23 @@ def inpaint(image, width, height, overlap_percentage, num_inference_steps, resiz
     # mask_blur = mask.filter(ImageFilter.GaussianBlur(10))
     # cnet_image = Image.composite(result, cnet_image, mask)
     size    = 10; # how many pixel extension do you want to dilate
-    number  = 1;  # numbering purpose 
-    trimap_mask = trimap(image, name, size, number, erosion=False)
+    number  = 1;  # numbering purpose
+    trimap_mask = trimap(np.array(mask), name, size, number, erosion=False)
 
     # estimate alpha from image and trimap
-    alpha = estimate_alpha_cf(cnet_image, trimap_mask)
+    alpha = estimate_alpha_cf(np.array(cnet_image), trimap_mask)
     
     # make gray background
     background = result
     
     # estimate foreground from image and alpha
-    foreground = estimate_foreground_ml(cnet_image, alpha)
+    foreground = estimate_foreground_ml(np.array(cnet_image), alpha)
     
     # blend foreground with background and alpha, less color bleeding
-    new_image = blend(foreground, background, alpha)
+    new_image = blend(foreground, np.array(background), alpha)
     # cnet_image.paste(result, (0, 0), mask)
 
-    return new_image, background
+    return Image.fromarray(new_image), background
 
 def preview_image_and_mask(image, width, height, overlap_percentage, resize_option, custom_resize_percentage, alignment, overlap_left, overlap_right, overlap_top, overlap_bottom):
     background, mask = prepare_image_and_mask(image, width, height, overlap_percentage, resize_option, custom_resize_percentage, alignment, overlap_left, overlap_right, overlap_top, overlap_bottom)