CharlieAmalet commited on
Commit
956147e
·
verified ·
1 Parent(s): c75dec5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -6
app.py CHANGED
@@ -21,14 +21,10 @@ css = """
21
  }
22
 
23
  """
24
- DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
 
25
  model = torch.hub.load('isl-org/ZoeDepth', "ZoeD_N", pretrained=True).to("cpu").eval()
26
 
27
- # config_mode="infer"
28
- # pretrained_resource = f"local::C:/Users/Charl/.cache/torch/hub/checkpoints/ZoeD_M12_N.pt"
29
- # config = get_config("zoedepth", config_mode, pretrained_resource=pretrained_resource)
30
- # model = build_model(config).to(DEVICE).eval()
31
-
32
  # title = "# ZoeDepth"
33
  # description = """Official demo for **ZoeDepth: Zero-shot Transfer by Combining Relative and Metric Depth**."""
34
 
 
21
  }
22
 
23
  """
24
+ # DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
25
+ DEVICE = 'cuda'
26
  model = torch.hub.load('isl-org/ZoeDepth', "ZoeD_N", pretrained=True).to("cpu").eval()
27
 
 
 
 
 
 
28
  # title = "# ZoeDepth"
29
  # description = """Official demo for **ZoeDepth: Zero-shot Transfer by Combining Relative and Metric Depth**."""
30