train_model.py

pip install transformers datasets evaluate accelerate