This is the code and data used for the example explained in my medium article you can read here: https://medium.com/@adrian.errea.lopez/from-pytorch-model-to-mobile-application-50bc5729ed83
Enjoy 😊
PD: Kudos to Salmen Bejaoui in Unsplash for the Image used