Update app.py

app.py

@@ -580,7 +580,15 @@ def draw_mask(l, t, th, v, d, evt: gr.EventData):
       dilation = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (15-(t*2+1), 15-(t*2+1)), (t, t))
       mask = cv2.dilate(mask, dilation)
 
-      bd = cv2.convertScaleAbs(cv2.Laplacian(bg, cv2.CV_64F, 5))
+      #bd = cv2.convertScaleAbs(cv2.Laplacian(bg, cv2.CV_64F, 5))
+        
+      n = int((bg.shape[0]*bg.shape[1]) / (256*256))
+      _bg = cv2.medianBlur(bg, 255)
+    
+      for i in range(0, n):
+          _bg = cv2.medianBlur(_bg, 255)
+        
+      bg = cv2.convertScaleAbs(bg.astype(np.int16)-_bg.astype(np.int16)).astype(np.uint8)
 
       #indices = np.arange(0,256)   # List of all colors 
       #divider = np.linspace(0,255,l+1)[1] # we get a divider