Update app.py

app.py

@@ -54,6 +54,13 @@ DAMV2_configs = {
     'vitl': {'encoder': 'vitl', 'features': 256, 'out_channels': [256, 512, 1024, 1024]},
 }
 encoder = 'vitb' # or 'vits', 'vitb', 'vitl'
+encoder2name = {
+    'vits': 'Small',
+    'vitb': 'Base',
+    'vitl': 'Large',
+    'vitg': 'Giant', # we are undergoing company review procedures to release our giant model checkpoint
+}
+model_name = encoder2name[encoder]
 DAMV2 = DepthAnythingV2(**DAMV2_configs[encoder])
 filepath = hf_hub_download(repo_id=f"depth-anything/Depth-Anything-V2-{model_name}", filename=f"depth_anything_v2_{encoder}.pth", repo_type="model")
 state_dict = torch.load(filepath, map_location="cpu")