Update app.py

app.py

@@ -576,6 +576,13 @@ def draw_mask(l, t, v, d, evt: gr.EventData):
     if len(edge) == 0 or params["fnum"] != frame_selected or params["l"] != l:
       if len(edge) > 0:
           d["background"] = cv2.imread(depths[frame_selected]).astype(np.uint8)
+          
+          if d["background"].shape[0] == 2048: #height
+            gradient = cv2.imread('./gradient_large.png').astype(np.uint8)
+          elif d["background"].shape[0] == 1024:
+            gradient = cv2.imread('./gradient.png').astype(np.uint8)
+          else:
+            gradient = cv2.imread('./gradient_small.png').astype(np.uint8)
 
       bg = cv2.cvtColor(d["background"], cv2.COLOR_RGBA2GRAY)