zino36 commited on
Commit
34db981
·
verified ·
1 Parent(s): 055cd1b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -54,7 +54,7 @@ filepath = hf_hub_download(repo_id="depth-anything/Depth-Anything-V2-Metric-Indo
54
  #model.load_state_dict(state_dict)
55
  #model = model.to(DEVICE).eval()
56
 
57
- pipe = pipeline(task="depth-estimation", model="depth-anything/Depth-Anything-V2-Metric-Indoor-Large-hf", device=0) # Use appropriate device
58
 
59
  title = "# Depth Anything V2"
60
  description = """Official demo for **Depth Anything V2**.
 
54
  #model.load_state_dict(state_dict)
55
  #model = model.to(DEVICE).eval()
56
 
57
+ pipe = pipeline(task="depth-estimation", model="depth-anything/Depth-Anything-V2-Metric-Indoor-Large-hf", device=DEVICE)
58
 
59
  title = "# Depth Anything V2"
60
  description = """Official demo for **Depth Anything V2**.