prediction from trained and saved model works