Tflite on Android: resulting classification list contains only one category

Question

I have a simple image classification model that works just fine in Python, it has 3 categories.

model = tf.keras.Sequential([

  augmentation,
  layers.experiment