Alternate Application for this UNET

Hi all,

My application is not edge detection but reproducing a grayscale image. I am using this network as is, and it is giving me a black or white edge image. Will I be able to use this UNET for my application, and if so, please advise. If not, is there any other UNET or network I can use for my situation, and can you please point me to that source?

Thanks