Possible to have Multilabel classifier in CoreML format (for iOS app)?

After exporting a multilabel classifier model with around 407 labels after lesson 3, I wanted to put it into an iOS app.

However, when I converted it from Pytorch .pkl model to -> ONNX to -> CoreML and tried it out an error stated that the number of classes (which I got directly from the learner.data.classes) didn’t equal the size of the output layer. Is this because a muli Anyone been able to do this?