Update app.py

app.py

@@ -55,9 +55,10 @@ model_configs = {
     'vitl': {'encoder': 'vitl', 'features': 256, 'out_channels': [256, 512, 1024, 1024]},
     'vitg': {'encoder': 'vitg', 'features': 384, 'out_channels': [1536, 1536, 1536, 1536]}
 }
+
 encoder = 'vitl'
 model = DepthAnythingV2(**model_configs[encoder])
-state_dict = torch.load(f'depth_anything_v2_{encoder}.pth', map_location="cpu")
+state_dict = torch.load(f'/home/user/app/checkpoints/depth_anything_v2_{encoder}.pth', map_location="cpu")
 model.load_state_dict(state_dict)
 model = model.to(DEVICE).eval()