Code structuring for text classification with hf bert-uncase

[bvng]

I appreciate the feedbacks, I’ll look into it to improve