Hello Could you please write on the ReadMe some instructions on how to train the model? Thank you