Update app.py

app.py

@@ -189,10 +189,6 @@ def remove_bg(fl, count, mh, ms, md, lm):
         fr_shadow = np.bitwise_and(fr_diff, fr_not).astype(np.uint8)
         fr_fg = np.bitwise_or(fr_diff, fr_mask).astype(np.uint8)
         cv2.imwrite(fl_, fr_mask)
-
-        fr_fg[fr_fg>0] = 3 #probable fg
-        mask, bgdModel, fgdModel = cv2.grabCut(fr, fr_fg, None,None,None,15, cv2.GC_INIT_WITH_MASK)
-        mask = np.where((mask==2)|(mask==0),0,1).astype('uint8')
         
         m = cv2.inRange(fr, np.array([240,240,240]), np.array([255,255,255]))
         fr[m>0] = (239,239,239)
@@ -200,6 +196,10 @@ def remove_bg(fl, count, mh, ms, md, lm):
         fr[m>0] = (16,16,16)
         fr[fr_shadow>0] = (fr[fr_shadow>0] / 17).astype(np.uint8)
         #fr[fr_fg==0] = (255,255,255)
+
+        fr_fg[fr_fg>0] = 3 #probable fg
+        mask, bgdModel, fgdModel = cv2.grabCut(fr, fr_fg, None,None,None,31, cv2.GC_INIT_WITH_MASK)
+        mask = np.where((mask==2)|(mask==0),0,1).astype('uint8')
         fr[mask==0] = (255,255,255)
         
         cv2.imwrite(fl, fr)