Update app.py

app.py

@@ -148,8 +148,6 @@ def linscale(linear):
 def remove_bg(fl, count, mh, ms, md, lm, backSub):
     global fl_
     fr = cv2.imread(fl).astype(np.uint8)
-
-    fgMask = backSub.apply(fr)
     
     #b = 3
     #element = cv2.getStructuringElement(cv2.MORPH_RECT, (2 * b + 1, 2 * b + 1), (b, b))
@@ -202,6 +200,7 @@ def remove_bg(fl, count, mh, ms, md, lm, backSub):
         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')
+        fgMask = backSub.apply(fr)
         fr[fgMask==0] = (255,255,255)
         
         cv2.imwrite(fl, fr)