Almost there!

Clone this project to retrain this model.