Very nice, thank you for writing the article and sharing it!
I noticed that you are using Transformers 2.9.0. Have you tried doing the same in more recent versions where a lot of the interface has changed? If so, were you successful or was that the reason you went back to an older version?