Spaces:

wondervictor
/

ControlAR

Runtime error

wondervictor commited on Nov 4, 2024

Commit

3094fff

verified ·

1 Parent(s): 3659749

Update condition/midas/depth.py

Files changed (1) hide show

condition/midas/depth.py CHANGED Viewed

@@ -85,6 +85,7 @@ def load_model(model_type):
     # https://github.com/isl-org/MiDaS/blob/master/run.py
     # load network
     model_path = ISL_PATHS[model_type]
     if model_type == "dpt_large":  # DPT-Large
         model = DPTDepthModel(
             path=model_path,

     # https://github.com/isl-org/MiDaS/blob/master/run.py
     # load network
     model_path = ISL_PATHS[model_type]
+    model_path = 'checkpoints/dpt_hybrid-midas-501f0c75.pt'
     if model_type == "dpt_large":  # DPT-Large
         model = DPTDepthModel(
             path=model_path,