The pre-trained model should be place here as ./pretrain_models/model_4.pth.

If you do not have the pre-trained model file, you can get it from the Code Ocean capsule.