I've figured it out, you need to train and compile the model using python 3.9 and tensorflow 2.8 as the latest flutter tensorflow lite lib doesn't support some operations that are later on was added to tensorflow lite