Spaces:

Tej3
/

DepthEstimation

Running

Tej3 commited on Apr 28, 2023

Commit

9a6115d

1 Parent(s): 92f6568

fixing path errors

Files changed (1) hide show

app.py CHANGED Viewed

@@ -72,7 +72,7 @@ Models = ['Resnet_enc','Unet','Densenet_enc']
 for location in location_types:
     for model in Models:
         ckpt_dir = f"{CWD}/ckpt/{CKPT_FILE_NAMES[location][model]}"
-        load_model(CKPT_FILE_NAMES[location][model], MODEL_CLASSES[location][model])

 for location in location_types:
     for model in Models:
         ckpt_dir = f"{CWD}/ckpt/{CKPT_FILE_NAMES[location][model]}"
+        load_model(ckpt_dir, MODEL_CLASSES[location][model])