Update app.py

app.py

@@ -297,8 +297,8 @@ def blur_image(image, depth, blur_data):
     j = 0
     while j < 256:
         i = 255 - j
-        blur_lo = [i,i,i]
-        blur_hi = [i+1,i+1,i+1]
+        blur_lo = np.array([i,i,i])
+        blur_hi = np.array([i+1,i+1,i+1])
         blur_mask = cv2.inRange(depth, blur_lo, blur_hi)
         
         print(f'kernel size {int(blur_a[j])}')
@@ -341,7 +341,9 @@ def draw_mask(v, d, evt: gr.EventData):
     w = int(vv[2])
     h = int(vv[3])
 
+    d_copy = d["background"].copy()
     rgb = cv2.cvtColor(d["background"], cv2.COLOR_RGBA2RGB)
+    d["background"] = cv2.applyColorMap(cv2.cvtColor(d["background"], cv2.COLOR_RGBA2GRAY), cv2.COLORMAP_HSV)
     mask = np.zeros(rgb.shape[:2], np.uint8)
     bgdModel = np.zeros((1,65), np.float64)
     fgdModel = np.zeros((1,65), np.float64)