Blocked on unet_model for U-net assignment

Here is a thread that discusses a similar issue: