ZenosArrows commited on
Commit
3d8b7f4
·
verified ·
1 Parent(s): 7847c9b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -72,10 +72,10 @@ def infer_video_depth(
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):
 
72
  grayscale: bool = False,
73
  output_dir: str = './outputs',
74
  input_size: int = 518,
75
+ progress=gr.Progress(track_tqdm=True)
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)
79
 
80
  video_name = os.path.basename(input_video)
81
  if not os.path.exists(output_dir):