Hi Aruna,
the source code is at the Model Garden repo here 4
One question, are you planning on using this model on a mobile device? I’m asking because it might be very slow! this model is very complex and has a big input (1024x1024). It’s not really a model for on-device inference. There’re many better options (good tutorial here 1).