Update app.py

app.py

@@ -69,6 +69,7 @@ def infer_video_depth(
     max_len: int = -1,
     target_fps: int = -1,
     max_res: int = 1280,
+    grayscale: bool = False,
     output_dir: str = './outputs',
     input_size: int = 518,
 ):
@@ -82,7 +83,7 @@ def infer_video_depth(
     processed_video_path = os.path.join(output_dir, os.path.splitext(video_name)[0]+'_src.mp4')
     depth_vis_path = os.path.join(output_dir, os.path.splitext(video_name)[0]+'_vis.mp4')
     save_video(frames, processed_video_path, fps=fps)
-    save_video(depths, depth_vis_path, fps=fps, is_depths=True)
+    save_video(depths, depth_vis_path, fps=fps, is_depths=True, grayscale=grayscale)
 
     gc.collect()
     torch.cuda.empty_cache()
@@ -145,6 +146,10 @@ def construct_demo():
                             value=1280,
                             step=1,
                         )
+                        grayscale = gr.Checkbox(
+                            label="grayscale",
+                            value=False,
+                        )
                     generate_btn = gr.Button("Generate")
             with gr.Column(scale=2):
                 pass
@@ -168,7 +173,8 @@ def construct_demo():
                 input_video,
                 max_len,
                 target_fps,
-                max_res
+                max_res,
+                grayscale
             ],
             outputs=[processed_video, depth_vis_video],
         )