Janeka commited on
Commit
5dadb9e
·
verified ·
1 Parent(s): 71b4f1b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -3,6 +3,7 @@ from PIL import Image
3
  from rembg import new_session, remove
4
  from skimage import filters
5
  import gradio as gr
 
6
 
7
  # Initialize models
8
  isnet_session = new_session("isnet-general-use")
@@ -24,7 +25,11 @@ def perfect_remove_bg(img):
24
  footprint=np.ones((3,3), np.uint8)
25
  )
26
 
27
- # U²Net mask (FIXED - now has all parentheses closed)
 
 
 
 
28
  u2net_result = remove(img, session=u2net_session)
29
  u2net_mask = np.array(u2net_result.split()[-1])
30
 
 
3
  from rembg import new_session, remove
4
  from skimage import filters
5
  import gradio as gr
6
+ import cv2 # Added for dilation and GaussianBlur
7
 
8
  # Initialize models
9
  isnet_session = new_session("isnet-general-use")
 
25
  footprint=np.ones((3,3), np.uint8)
26
  )
27
 
28
+ # Enhanced edge preservation
29
+ mask = cv2.dilate(mask, np.ones((1,1), np.uint8), iterations=1)
30
+ mask = cv2.GaussianBlur(mask, (3,3), 0)
31
+
32
+ # U²Net mask
33
  u2net_result = remove(img, session=u2net_session)
34
  u2net_mask = np.array(u2net_result.split()[-1])
35