Update train.py

train.py

@@ -1,4 +1,4 @@
-from transformers import MBartForSequenceClassification, GPT2Tokenizer, TrainingArguments, Trainer
+from transformers import MBartForSequenceClassification, BertTokenizer, TrainingArguments, Trainer
 from datasets import Dataset
 
 
@@ -27,7 +27,7 @@ def prepare_text_dataset(data, chunk_size):
 def init():
     # Load the model and tokenizer
     model_name = "language-ml-lab/AzerBert"  # Replace with your model name if different
-    tokenizer = GPT2Tokenizer.from_pretrained(model_name)
+    tokenizer = BertTokenizer.from_pretrained(model_name)
     model = MBartForSequenceClassification.from_pretrained(model_name)
     chunk_size = 512