Added model argument to set_classifier