This was a demonstration of how we can use AI models on an iOS device. Having said that, there are a lot of tasks that can be added to the existing app:

• Improve the UI design
• Add input checks to validate the data entered
• Host the Flask app (Heroku, AWS, and more)
• Publish the app

All these tasks are independent of our core AI theme, so they can be addressed as an exercise for the reader. The complete code and project files for both the Android and iOS apps are available as chapter2_android_prediction and chapter2_ios_prediction.