Creating inputs to the TensorFlow model