How do I change the predict function in an image classification android studio app to predict more than two classes?

Question

The app I have works well but when I changed the model to one that was trained on 12 classes I couldn\'t figure out how to change the code in a way that would predict more than