Delete train_model.py