I loaded the model on colab to share and tried to convert which gave me a converter model with the correct output details. So I wondered why I wasn’t in my machine, then I checked my TensorFlow version which was 2.4.1 as compared to the colabs’ 2.6.0 and so I upgraded it and it works fine now.