zino36 commited on
Commit
106e7e7
·
verified ·
1 Parent(s): 7f1457e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ css = """
26
  """
27
  DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
28
  encoder = 'vitl' # can also be 'vitb' or 'vitl'
29
- model = DepthAnything.from_pretrained(f"LiheYoung/depth_anything_{encoder}14").to(DEVICE).eval()
30
 
31
  title = "# Depth Anything"
32
  description = """Official demo for **Depth Anything: Unleashing the Power of Large-Scale Unlabeled Data**.
 
26
  """
27
  DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
28
  encoder = 'vitl' # can also be 'vitb' or 'vitl'
29
+ model = DepthAnything.from_pretrained("checkpoints_metric_depth/depth_anything_metric_depth_indoor.pt").to(DEVICE).eval()
30
 
31
  title = "# Depth Anything"
32
  description = """Official demo for **Depth Anything: Unleashing the Power of Large-Scale Unlabeled Data**.