How to do inference with RoBERTa on a multi classification task

[dxlong2000]

Hi the team,

Thank you for your great work. In the document "Finetuning RoBERTa on a custom classification task", I saw you provided the inference code for RoBERTa binary classification.

However, I would like to ask if it is possible that you provide the inference codes for multi-class classification with RoBERTa as well?

Thanks!