JohanDL commited on
Commit
ee7a6d5
1 Parent(s): 277dfb3
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ def make_video(video_path, outdir='./vis_video_depth',encoder='vitl'):
27
  margin_width = 50
28
 
29
  DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
30
-
31
  depth_anything = DepthAnything.from_pretrained('LiheYoung/depth_anything_{}14'.format(encoder)).to(DEVICE).eval()
32
 
33
  total_params = sum(param.numel() for param in depth_anything.parameters())
 
27
  margin_width = 50
28
 
29
  DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
30
+ DEVICE = "cuda"
31
  depth_anything = DepthAnything.from_pretrained('LiheYoung/depth_anything_{}14'.format(encoder)).to(DEVICE).eval()
32
 
33
  total_params = sum(param.numel() for param in depth_anything.parameters())