Remove the output of train_model.