Update train_model.py

train_model.py

@@ -74,7 +74,7 @@ def load_and_prepare_dataset(task, dataset_name, tokenizer, sequence_length):
         logging.info("Dataset loaded successfully.")
 
         def tokenize_function(examples):
-            return tokenizer(examples['text'], truncation=True, max_length=sequence_length)
+            return tokenizer(examples['text'], truncation=True, padding='max_length', max_length=sequence_length)
         
         # Tokenize the dataset
         tokenized_datasets = dataset.shuffle(seed=42).select(range(500)).map(tokenize_function, batched=True)