import os
os.system("pip install transformers datasets torch gradio")  # Install missing libraries

from transformers import Trainer, TrainingArguments
# Rest of your code...



from transformers import Trainer, TrainingArguments

training_args = TrainingArguments(
    output_dir="./results",
    num_train_epochs=3,
    per_device_train_batch_size=8,
    save_steps=10_000,
    save_total_limit=2,
)

trainer = Trainer(
    model=model,
    args=training_args,
    train_dataset="training_data.jsonl",
)

trainer.train()