Start by loading your model and specify the number of expected labels.