Thanks! Yes, I am. The intention is for the user to also be able to correct the output of the model.

After that I want to give the user an easy interface for training their own models (transfer learning using an existing base model).