JUGGHM commited on
Commit
4ded653
·
1 Parent(s): bb7b518

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -32,13 +32,13 @@ from mono.utils.unproj_pcd import reconstruct_pcd, save_point_cloud
32
  import gradio as gr
33
 
34
  os.chdir(CODE_SPACE)
35
- cfg = Config.fromfile('https://huggingface.co/JUGGHM/Metric3D/blob/main/mono/configs/_base_/models/encoder_decoder/dino_vit_small_reg.dpt_raft.py')
36
 
37
  torch.hub.download_url_to_file('https://images.unsplash.com/photo-1437622368342-7a3d73a34c8f', 'turtle.jpg')
38
  torch.hub.download_url_to_file('https://images.unsplash.com/photo-1519066629447-267fffa62d4b', 'lions.jpg')
39
 
40
  model = get_configured_monodepth_model(cfg, )
41
- model, _, _, _ = load_ckpt(cfg.load_from, model, strict_match=False)
42
  model.eval()
43
 
44
  device = "cpu"
 
32
  import gradio as gr
33
 
34
  os.chdir(CODE_SPACE)
35
+ cfg = Config.fromfile('./mono/configs/_base_/models/encoder_decoder/dino_vit_small_reg.dpt_raft.py')
36
 
37
  torch.hub.download_url_to_file('https://images.unsplash.com/photo-1437622368342-7a3d73a34c8f', 'turtle.jpg')
38
  torch.hub.download_url_to_file('https://images.unsplash.com/photo-1519066629447-267fffa62d4b', 'lions.jpg')
39
 
40
  model = get_configured_monodepth_model(cfg, )
41
+ model, _, _, _ = load_ckpt('./weight/metric_depth_vit_small_800k.pth.pth', model, strict_match=False)
42
  model.eval()
43
 
44
  device = "cpu"