Update app.py

app.py

@@ -80,15 +80,13 @@ def infer_video_depth(
     if not os.path.exists(output_dir):
         os.makedirs(output_dir)
 
-    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, grayscale=grayscale)
+    processed_video_path = os.path.join(output_dir, os.path.splitext(video_name)[0]+'.mp4')
+    save_video(frames, depths, processed_video_path, fps=fps, grayscale=grayscale)
 
     gc.collect()
     torch.cuda.empty_cache()
 
-    return [processed_video_path, depth_vis_path]
+    return processed_video_path
 
 
 def construct_demo():
@@ -112,14 +110,6 @@ def construct_demo():
                         show_share_button=True,
                         scale=5,
                     )
-                    depth_vis_video = gr.Video(
-                        label="Generated Depth Video",
-                        interactive=False,
-                        autoplay=True,
-                        loop=True,
-                        show_share_button=True,
-                        scale=5,
-                    )
 
         with gr.Row(equal_height=True):
             with gr.Column(scale=1):
@@ -162,7 +152,7 @@ def construct_demo():
                 target_fps,
                 max_res
             ],
-            outputs=[processed_video, depth_vis_video],
+            outputs=[processed_video],
             fn=infer_video_depth,
             cache_examples="lazy",
         )
@@ -176,7 +166,7 @@ def construct_demo():
                 max_res,
                 grayscale
             ],
-            outputs=[processed_video, depth_vis_video],
+            outputs=[processed_video],
         )
 
     return demo