JohanDL commited on
Commit
0c01bdf
1 Parent(s): 2abae72
Files changed (1) hide show
  1. app.py +1 -1
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":"big","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
 
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