BERT model for text category classification