A model for the nonlinear rheology of associating polymers