Update app.py

app.py

@@ -33,9 +33,15 @@ model_configs = {
     'vitl': {'encoder': 'vitl', 'features': 256, 'out_channels': [256, 512, 1024, 1024]},
     'vitg': {'encoder': 'vitg', 'features': 384, 'out_channels': [1536, 1536, 1536, 1536]}
 }
+encoder2name = {
+    'vits': 'Small',
+    'vitb': 'Base',
+    'vitl': 'Large',
+    'vitg': 'Gaint',
+}
 encoder = 'vitg'
 model = DepthAnythingV2(**model_configs[encoder])
-filepath = hf_hub_download(repo_id="LiheYoung/Depth-Anything-V2-Checkpoints", filename=f"depth_anything_v2_{encoder}.pth", repo_type="model")
+filepath = hf_hub_download(repo_id=f"depth-anything/Depth-Anything-V2-{encoder2name[encoder]}", filename=f"model.pth", repo_type="model")
 state_dict = torch.load(filepath, map_location="cpu")
 model.load_state_dict(state_dict)
 model = model.to(DEVICE).eval()