Prediction of drag reduction with a viscoelastic model