add model to the args for segment_label