If you are intending to process a single image at a time, the input shape should probably be 1x150528. You can do something like image = image.reshape(1, 150528). Then if you change the input size of the model to 150528 it should work.