Update app.py

app.py

@@ -164,9 +164,9 @@ def remove_bg(fl, count, md):
     fr_diff = cv2.cvtColor(fr_diff, cv2.COLOR_BGR2GRAY)
     
     #md = 12
-    fg = np.bitwise_or(hsv[:,:,2]>=md, hsv[:,:,1]>=md, hsv[:,:,0]>=md)
-    fr_diff[fg>0] = 255
-    fr_diff[fg==0] = 0
+    bg = cv2.inRange(hsv, np.array([0,0,0]), np.array([180,md,md]))
+    fr_diff[bg>0] = 0
+    fr_diff[bg==0] = 255
     
     cv2.rectangle(fr_diff,(0,0),(fr_diff.shape[1]-1,fr_diff.shape[0]-1),(255,255,255),1)
     mask = cv2.floodFill(fr_diff, None, (0, 0), 255, 0, 0, (4 | cv2.FLOODFILL_FIXED_RANGE))[2] #(4 | cv.FLOODFILL_FIXED_RANGE | cv.FLOODFILL_MASK_ONLY | 255 << 8)