Update app.py

app.py

@@ -371,8 +371,8 @@ def draw_mask(f, n, c, k, v, d, evt: gr.EventData):
 
     diff = (bg-cv2.cvtColor(gradient, cv2.COLOR_RGBA2GRAY)).astype(np.uint8)
     mask = cv2.inRange(diff, -int(c/2), int(c/2))
-    kernel = np.ones((k,k),np.float32)/(k*k)
-    mask = cv2.filter2D(mask,-1,kernel)
+    #kernel = np.ones((k,k),np.float32)/(k*k)
+    #mask = cv2.filter2D(mask,-1,kernel)
 
     indices = np.arange(0,256)   # List of all colors 
     divider = np.linspace(0,255,n+1)[1] # we get a divider
@@ -382,7 +382,7 @@ def draw_mask(f, n, c, k, v, d, evt: gr.EventData):
 
     #for i in range(n):
     #    bg[(bg >= i*255/n) & (bg < (i+1)*255/n)] = i*255/(n-1)
-    bg = cv2.convertScaleAbs(palette[bg]).astype(np.uint8) # Converting image back to uint
+    bg = cv2.convertScaleAbs(palette[bg] + mask).astype(np.uint8) # Converting image back to uint
 
     scale = 1
     delta = 0