Update app.py

app.py

@@ -196,8 +196,11 @@ def remove_bg(fl, s, l, v):
     m_ = cv2.inRange(frame_, np.array([1,1,1]), np.array([127,127,127]))
     frame_[m_>0] = (127,127,127)
 
-
-    
+    frame_ = 255 - frame_
+    m_ = cv2.inRange(frame_, np.array([255,255,255]), np.array([255,255,255]))
+    frame_[m_>0] = (128,128,128)
+    m_ = cv2.inRange(frame_, np.array([127,127,127]), np.array([127,127,127]))
+    frame_[m_>0] = (255,255,255)
 
     cv2.imwrite(fl, frame_)
     return fl