Spaces:

jho
/

MonocularDepth

Build error

ohjho commited on Jun 20, 2022

Commit

a90faa2

•

1 Parent(s): b240372

fource reload

Files changed (1) hide show

DPT.py CHANGED Viewed

@@ -11,7 +11,7 @@ MODEL_DICT = {
 def load_model(model_type = 'DPT_Large'):
     assert model_type in MODEL_DICT.keys(), f'{model_type} is not a valid model_type: {MODEL_DICT.keys()}'
-    midas = torch.hub.load("intel-isl/MiDaS", model_type)
     device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
     print(f'---DPT will use device: {device}')

 def load_model(model_type = 'DPT_Large'):
     assert model_type in MODEL_DICT.keys(), f'{model_type} is not a valid model_type: {MODEL_DICT.keys()}'
+    midas = torch.hub.load("intel-isl/MiDaS", model_type, force_reload=True)
     device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
     print(f'---DPT will use device: {device}')