Predicting balls and strikes using TensorFlow.js