JohanDL commited on
Commit
cbdb616
1 Parent(s): 431dd55

Update app

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -32,8 +32,8 @@ def make_video(video_path, outdir='./vis_video_depth',encoder='vitl'):
32
  # depth_anything = DepthAnything.from_pretrained('LiheYoung/depth_anything_{}14'.format(encoder)).to(DEVICE).eval()
33
  depth_anything = pipeline(task = "depth-estimation", model="nielsr/depth-anything-small", device=0)
34
 
35
- total_params = sum(param.numel() for param in depth_anything.parameters())
36
- print('Total parameters: {:.2f}M'.format(total_params / 1e6))
37
 
38
  transform = Compose([
39
  Resize(
 
32
  # depth_anything = DepthAnything.from_pretrained('LiheYoung/depth_anything_{}14'.format(encoder)).to(DEVICE).eval()
33
  depth_anything = pipeline(task = "depth-estimation", model="nielsr/depth-anything-small", device=0)
34
 
35
+ # total_params = sum(param.numel() for param in depth_anything.parameters())
36
+ # print('Total parameters: {:.2f}M'.format(total_params / 1e6))
37
 
38
  transform = Compose([
39
  Resize(