Update app.py

app.py

@@ -70,6 +70,7 @@ def get_masked_img(img, w, h, features, orig_h, orig_w, input_h, input_w, dilate
 
     masks = masks.astype(np.uint8) * 255
 
+    dilate_kernel_size = 20
     # dilate mask to avoid unmasked edge effect
     if dilate_kernel_size is not None:
         masks = [dilate_mask(mask, dilate_kernel_size) for mask in masks]