Update train.py

train.py

@@ -26,7 +26,7 @@ def prepare_text_dataset(data, chunk_size):
 
 def init():
     # Load the model and tokenizer
-    model_name = "facebook/bart-base-multilingual"  # Replace with your model name if different
+    model_name = "language-ml-lab/AzerBert"  # Replace with your model name if different
     tokenizer = GPT2Tokenizer.from_pretrained(model_name)
     model = MBartForSequenceClassification.from_pretrained(model_name)
     chunk_size = 512