Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 |
|