Update README.md

README.md

@@ -22,7 +22,7 @@ This model is an unconditional MIDI classic piano music generator trained on the
 To use this model and generate your own music, just download `model.pt` and `use.py` and run it with Python3. Have fun :D
 
 ## Training code
-If you want to have the training code for this, just ask me :-)
+Use `train.py`.
 
 ## Training data
 We trained on [https://storage.googleapis.com/magentadata/datasets/maestro/v3.0.0/maestro-v3.0.0-midi.zip](https://storage.googleapis.com/magentadata/datasets/maestro/v3.0.0/maestro-v3.0.0-midi.zip).