ZenosArrows commited on
Commit
cfe841b
·
verified ·
1 Parent(s): b528de6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -72,9 +72,10 @@ def infer_video_depth(
72
  grayscale: bool = False,
73
  output_dir: str = './outputs',
74
  input_size: int = 518,
 
75
  ):
76
  frames, target_fps = read_video_frames(input_video, max_len, target_fps, max_res)
77
- depths, fps = video_depth_anything.infer_video_depth(frames, target_fps, input_size=input_size, device=DEVICE)
78
 
79
  video_name = os.path.basename(input_video)
80
  if not os.path.exists(output_dir):
 
72
  grayscale: bool = False,
73
  output_dir: str = './outputs',
74
  input_size: int = 518,
75
+ progress=gr.Progress()
76
  ):
77
  frames, target_fps = read_video_frames(input_video, max_len, target_fps, max_res)
78
+ depths, fps = video_depth_anything.infer_video_depth(frames, target_fps, input_size=input_size, device=DEVICE, progress=progress.tqdm)
79
 
80
  video_name = os.path.basename(input_video)
81
  if not os.path.exists(output_dir):