changed input file name in training.js