Create app.py

app.py

@@ -0,0 +1,14 @@
+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')