Fix dataset reference error

spanish_medica_llm.py

@@ -525,7 +525,7 @@ def configAndRunTraining(basemodel, dataset, eval_dataset, tokenizer):
         
         trainer = Trainer(
                      model= basemodel,
-                     train_dataset = dataset['train'],
+                     train_dataset = dataset,
                      eval_dataset = eval_dataset,
                      args = training_args,
                      data_collator = data_collator_pretrain