This makes me skeptical. This is a key action - 1. Import, 2. Categorise ... - and it seemed like it received little thought from the developer. It's not a bug to fix mate, it's a core user flow that's very unthoughtful.
Fair, I'll ship a few improvements and continue iterating with users.
Great! Good luck to you with the launch. The main idea of on-device models is a very good one.