Update train_model.py

train_model.py

@@ -65,7 +65,7 @@ def load_and_prepare_dataset(task, dataset_name, tokenizer, sequence_length):
     logging.info(f"Loading dataset '{dataset_name}' for task '{task}'...")
     try:
         if task == "generation":
-            dataset = load_dataset(dataset_name, split='train')
+            dataset = load_dataset(dataset_name, split='train',use_auth_token=True)
             logging.info("Dataset loaded successfully for generation task.")
             def tokenize_function(examples):
                 return tokenizer(examples['text'], truncation=True, max_length=sequence_length)