Loading trained model to make prediction of single image

3

My model code is here incase that helps! UNET model on GPU runtime error