Android: TFLite giving giving wrong predictions compared to Python for same model

Question

I am creating an Android app which requires running a tensorflow image classification model. I created the following simple model in Python:

model = tf.keras.