Update app.py

app.py

@@ -178,8 +178,9 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
         global frames
         frames = orig_frames
         depths = depth_frames
-        
-        if masks[len(masks)-1].shape[0] == 4096:
+
+        print(depth_color.shape[0])
+        if depth_color.shape[0] == 4096:
             masks[len(masks)-1] = cv2.imread('gradient_large.png')
             depth_frames[len(masks)-1] = cv2.imread('gradient_large.png')
         else: