Feature Request: Add UNet model with documentation

[Namangarg110]

It would be great to add UNet to the repo as UNet is one of the best models for image segmentation.

I would be really interested in working on the same asap.

looking forward to your reply