Running the model on a browser using TensorFlow.js