from transformers import Trainer, TrainingArguments

def build_trainer(model, dataset, tokenizer, hub_model_id):
    print("⚙️ Initialisation du Trainer...")
    args = TrainingArguments(
        output_dir="./output",
        per_device_train_batch_size=2,
        num_train_epochs=1,
        save_steps=500,
        learning_rate=2e-5,
        logging_dir="./logs",
        push_to_hub=True,
        hub_model_id=hub_model_id
    )

    trainer = Trainer(
        model=model,
        args=args,
        train_dataset=dataset,
        tokenizer=tokenizer
    )
    return trainer