Documentation - for inference code in train.py to deploy with huggingface estimator

[sankalp-wns]

I am unable to find the documentation that describes how to write the inference code in train.py. I am able to train the model using huggingface estimator and deploy to sagemaker endpoint but I don't know how to add my inference code to train.py file. This is the reference.