freealise commited on
Commit
c57397b
1 Parent(s): cd9bbfc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -33,7 +33,6 @@ def make_video(video_path, outdir='./vis_video_depth',encoder='vitl'):
33
  to_tensor_transform = transforms.ToTensor()
34
 
35
  DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
36
- DEVICE = "cuda"
37
  # depth_anything = DepthAnything.from_pretrained('LiheYoung/depth_anything_{}14'.format(encoder)).to(DEVICE).eval()
38
  depth_anything = pipeline(task = "depth-estimation", model=f"nielsr/depth-anything-{mapper[encoder]}", device=0)
39
 
 
33
  to_tensor_transform = transforms.ToTensor()
34
 
35
  DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
 
36
  # depth_anything = DepthAnything.from_pretrained('LiheYoung/depth_anything_{}14'.format(encoder)).to(DEVICE).eval()
37
  depth_anything = pipeline(task = "depth-estimation", model=f"nielsr/depth-anything-{mapper[encoder]}", device=0)
38