Spaces:
Running
on
Zero
Running
on
Zero
Update condition/midas/depth.py
Browse files- condition/midas/depth.py +1 -0
condition/midas/depth.py
CHANGED
|
@@ -85,6 +85,7 @@ def load_model(model_type):
|
|
| 85 |
# https://github.com/isl-org/MiDaS/blob/master/run.py
|
| 86 |
# load network
|
| 87 |
model_path = ISL_PATHS[model_type]
|
|
|
|
| 88 |
if model_type == "dpt_large": # DPT-Large
|
| 89 |
model = DPTDepthModel(
|
| 90 |
path=model_path,
|
|
|
|
| 85 |
# https://github.com/isl-org/MiDaS/blob/master/run.py
|
| 86 |
# load network
|
| 87 |
model_path = ISL_PATHS[model_type]
|
| 88 |
+
model_path = 'checkpoints/dpt_hybrid-midas-501f0c75.pt'
|
| 89 |
if model_type == "dpt_large": # DPT-Large
|
| 90 |
model = DPTDepthModel(
|
| 91 |
path=model_path,
|