If you want a classification model you can use tensorLite and do the next code. but with this you should create the model in the web and then export. I want to create a mod