Update app.py

app.py

@@ -182,7 +182,7 @@ def remove_bg(fl, count):
         fr[m>0] = (239,239,239)
         m = cv2.inRange(fr, np.array([0,0,0]), np.array([15,15,15]))
         fr[m>0] = (16,16,16)
-        fr[fr_mask>0] = (fr[fr_mask>0] / 17).astype(np.uint8)
+        fr[fr_mask==255] = (fr[fr_mask==255] / 17).astype(np.uint8)
         fr[fr_diff==0] = (255,255,255)
         
         cv2.imwrite(fl, fr)