Add tokens per sentence option to counts_from_annotations