Spaces:
Running
on
Zero
Running
on
Zero
Update
Browse files
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())
|