Update app.py

app.py

@@ -343,18 +343,18 @@ def draw_mask(v, d, evt: gr.EventData):
 
     scale = 1
     delta = 0
-    ddepth = cv.CV_16S
+    ddepth = cv2.CV_16S
 
-    gray = cv.cvtColor(d["background"], cv.COLOR_BGR2GRAY)
-    grad_x = cv.Sobel(gray, ddepth, 1, 0, ksize=5, scale=scale, delta=delta, borderType=cv.BORDER_DEFAULT)
+    gray = cv2.cvtColor(d["background"], cv2.COLOR_BGR2GRAY)
+    grad_x = cv2.Sobel(gray, ddepth, 1, 0, ksize=5, scale=scale, delta=delta, borderType=cv2.BORDER_DEFAULT)
     # Gradient-Y
-    # grad_y = cv.Scharr(gray,ddepth,0,1)
-    grad_y = cv.Sobel(gray, ddepth, 0, 1, ksize=5, scale=scale, delta=delta, borderType=cv.BORDER_DEFAULT)
-    abs_grad_x = cv.convertScaleAbs(grad_x)
-    abs_grad_y = cv.convertScaleAbs(grad_y)
+    # grad_y = cv2.Scharr(gray,ddepth,0,1)
+    grad_y = cv2.Sobel(gray, ddepth, 0, 1, ksize=5, scale=scale, delta=delta, borderType=cv2.BORDER_DEFAULT)
+    abs_grad_x = cv2.convertScaleAbs(grad_x)
+    abs_grad_y = cv2.convertScaleAbs(grad_y)
 
-    grad = cv.addWeighted(abs_grad_x, 0.5, abs_grad_y, 0.5, 0)
-    d["composite"] = cv.cvtColor(255-grad, cv.COLOR_GRAY2RGB)
+    grad = cv2.addWeighted(abs_grad_x, 0.5, abs_grad_y, 0.5, 0)
+    d["composite"] = cv2.cvtColor(255-grad, cv2.COLOR_GRAY2RGB)
     
     return gr.ImageEditor(value=d)