Updated the new model docs to better reflect the updated custom_model implementation.