Spaces:
Running
Running
Update app.py
Browse files
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 |
|