JohanDL commited on
Commit
3bf2d11
1 Parent(s): 8c68c99

update app

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ def predict_depth(model, image):
19
 
20
  def make_video(video_path, outdir='./vis_video_depth',encoder='vitl'):
21
  if encoder not in ["vitl","vitb","vits"]:
22
- encoder = "vitl"
23
  # DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
24
  # model = DepthAnything.from_pretrained('LiheYoung/depth_anything_vitl14').to(DEVICE).eval()
25
  # Define path for temporary processed frames
 
19
 
20
  def make_video(video_path, outdir='./vis_video_depth',encoder='vitl'):
21
  if encoder not in ["vitl","vitb","vits"]:
22
+ encoder = "vits"
23
  # DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
24
  # model = DepthAnything.from_pretrained('LiheYoung/depth_anything_vitl14').to(DEVICE).eval()
25
  # Define path for temporary processed frames