Update app.py

app.py

@@ -417,9 +417,9 @@ def show_json(txt):
 
 def seg_frame(newmask):
 
-    if depth.shape[0] == 2048: #height
+    if newmask.shape[0] == 2048: #height
         gd = cv2.imread('./gradient_large.png', cv2.IMREAD_GRAYSCALE).astype(np.uint8)
-    elif depth.shape[0] == 1024:
+    elif newmask.shape[0] == 1024:
         gd = cv2.imread('./gradient.png', cv2.IMREAD_GRAYSCALE).astype(np.uint8)
     else:
         gd = cv2.imread('./gradient_small.png', cv2.IMREAD_GRAYSCALE).astype(np.uint8)