How can I predict the flow field by the checkpoint

I want to get a predicted model for flow field, which can export the flow field by the message like mach number or pressure that I entered. But I could only find the code to train and test the model on this edge. How can I use the model to predict flow field?