Help with inference and material apply to shape

[mali-tintash]

I am using pre trained weights for inference of material using the example given in Readme.
But i can't find the point where I'd provide a model with shape parts to the network along with the image + mask. So I can apply the predicted material to the model.

Please guide