Added the trained neural network model