from transformers import BertTokenizer, BertForSequenceClassification
from fastai.text.all import Learner

# 1) KoBERT 토크나이저 불러오기
tokenizer = BertTokenizer.from_pretrained('monologg/kobert')

# 2) 모델 정의 및 fine-tune
model = BertForSequenceClassification.from_pretrained('monologg/kobert', num_labels=len(categories))
# ... (FastAI 데이터블록 & Trainer 설정)

# 3) 학습 후 저장
learn = Learner(dls, model, loss_func=CrossEntropyLossFlat(), metrics=accuracy)
learn.fine_tune(3)
learn.export('model_ko.pkl')