Update predict.js