Spaces:
Running
on
Zero
Running
on
Zero
update
Browse files
app.py
CHANGED
@@ -23,7 +23,7 @@ def make_video(video_path, outdir='./vis_video_depth',encoder='vitl'):
|
|
23 |
if encoder not in ["vitl","vitb","vits"]:
|
24 |
encoder = "vits"
|
25 |
|
26 |
-
mapper = {"vits":"small","vitb":"
|
27 |
# DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
|
28 |
# model = DepthAnything.from_pretrained('LiheYoung/depth_anything_vitl14').to(DEVICE).eval()
|
29 |
# Define path for temporary processed frames
|
|
|
23 |
if encoder not in ["vitl","vitb","vits"]:
|
24 |
encoder = "vits"
|
25 |
|
26 |
+
mapper = {"vits":"small","vitb":"base","vitl":"large"}
|
27 |
# DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
|
28 |
# model = DepthAnything.from_pretrained('LiheYoung/depth_anything_vitl14').to(DEVICE).eval()
|
29 |
# Define path for temporary processed frames
|