Is this the code train_style_transfer.py I should use? and How to apply the trained model on my dataset?