aryswisnu commited on
Commit
cfe8e4c
·
1 Parent(s): 2ff39a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -55,7 +55,7 @@ def extract_image(img, pos_prompts, neg_prompts, threshold, blur_radius=5):
55
 
56
  # apply Gaussian blur for feathering
57
  final_mask_img = Image.fromarray((final_mask * 255).astype(np.uint8), "L")
58
- final_mask_img = final_mask_img.filter(Image.GaussianBlur(radius=blur_radius))
59
  final_mask = np.array(final_mask_img) / 255
60
  final_mask = final_mask > threshold
61
 
 
55
 
56
  # apply Gaussian blur for feathering
57
  final_mask_img = Image.fromarray((final_mask * 255).astype(np.uint8), "L")
58
+ final_mask_img = final_mask_img.filter(ImageFilter.GaussianBlur(radius=blur_radius))
59
  final_mask = np.array(final_mask_img) / 255
60
  final_mask = final_mask > threshold
61