Update app.py

app.py

@@ -178,6 +178,14 @@ 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:
+            masks[len(masks)-1] = cv2.imread('gradient_large.png')
+            depth_frames[len(masks)-1] = cv2.imread('gradient_large.png')
+        else:
+            masks[len(masks)-1] = cv2.imread('gradient.png')
+            depth_frames[len(masks)-1] = cv2.imread('gradient.png')
+        
         return final_vid, final_zip, frames, masks[frame_selected] #output_path
 
 def depth_edges_mask(depth):