Update app.py

app.py

@@ -179,10 +179,10 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
         frames = orig_frames
         depths = depth_frames
 
-        if depth_color.shape[0] == 4096:
+        if depth_color.shape[0] == 2048: #height
             masks[len(masks)-1] = './gradient_large.png'
             depth_frames[len(masks)-1] = './gradient_large.png'
-        elif depth_color.shape[0] == 2048:
+        elif depth_color.shape[0] == 1024:
             masks[len(masks)-1] = './gradient.png'
             depth_frames[len(masks)-1] = './gradient.png'
         else: