Hugging Face's logo

Spaces:
JUGGHM
/
Metric3D
Running on T4

App Files Community
8
JUGGHM commited on
Commit
df87b80
•
1 Parent(s): b696ba4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ model_small = get_configured_monodepth_model(cfg_small, )
46
  model_small, _, _, _ = load_ckpt('./weight/metric_depth_vit_small_800k.pth', model_small, strict_match=False)
47
  model_small.eval()
48
 
49
- device = "cpu"
50
  model_large.to(device)
51
  model_small.to(device)
52
 
 
46
  model_small, _, _, _ = load_ckpt('./weight/metric_depth_vit_small_800k.pth', model_small, strict_match=False)
47
  model_small.eval()
48
 
49
+ device = "cuda"
50
  model_large.to(device)
51
  model_small.to(device)
52