Fixed error by passing train df in the model