Inference on NYU Depth doesnt work

[CanKeles5]

Hi, I am trying to do inference on the NYU Depth dataset. The model seems to predict a single value (5) for all the pixels. I loaded the "v1-5-pruned-emaonly.ckpt" model from HuggingFace. Also, to load the model I have to remove the "['model']" part in line 44 of test.py, as it throws an error.

I would really appreciate any help. Thanks.