Private sample code for running categorisation on the mT5X