add prediction of cnn