Update app.py
Browse files
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 = "
|
| 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 |
|