LiheYoung commited on
Commit
1afd902
1 Parent(s): 3c2541d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -42,7 +42,7 @@ encoder2name = {
42
  encoder = 'vitg'
43
  model_name = encoder2name[encoder]
44
  model = DepthAnythingV2(**model_configs[encoder])
45
- filepath = hf_hub_download(repo_id=f"depth-anything/Depth-Anything-V2-{model_name}", filename=f"depth_anything_v2_{model_name.lower()}.pth", repo_type="model")
46
  state_dict = torch.load(filepath, map_location="cpu")
47
  model.load_state_dict(state_dict)
48
  model = model.to(DEVICE).eval()
 
42
  encoder = 'vitg'
43
  model_name = encoder2name[encoder]
44
  model = DepthAnythingV2(**model_configs[encoder])
45
+ filepath = hf_hub_download(repo_id=f"depth-anything/Depth-Anything-V2-{model_name}", filename=f"depth_anything_v2_{encoder}.pth", repo_type="model")
46
  state_dict = torch.load(filepath, map_location="cpu")
47
  model.load_state_dict(state_dict)
48
  model = model.to(DEVICE).eval()