Once #11 is complete there will be a neural net loaded ready to be run for inference.

To use this we need to:

• Write an interface function to be called instead of the existing scheme that will implement the NN
  • Concatenate the data as required into the expected input format - requires help from @surbhigoel77
  • Transform this reshaped/concatenated data into a torch tensor
  • Run the NN
  • Deconcatenate the returned data into physical variables
  • Use the variables as appropriate in the ptend updates

Generation of output data should come after this.