image size with new model

[alexng88]

I re-retrain my model and once I put inside it said:

Cannot copy to a TensorFlowLite tensor (Image) with 602112 bytes from a Java Buffer with 150528 bytes.

1 x 224 x 224 x 3 x 4 will get correct result but now it misses x 4

how can I fix it?