Unable to use a different model

[ghost]

I'm trying to use a custom model with the following properties:

The original model has these properties instead:

In theory, this should not be a problem as tflite is supposed to accept any TFLite model.

However, this error is thrown

E/AndroidRuntime(29968): Caused by: java.lang.IllegalArgumentException: Cannot copy from a TensorFlowLite tensor (Identity) with shape [1, 3, 6400, 85] to a Java object with shape [1, 80]

when I try to classify an image

Tflite.runModelOnFrame(
      bytesList: image.planes.map((plane) => plane.bytes).toList(),
    )

A similar error

E/AndroidRuntime( 2360): Caused by: java.lang.IllegalArgumentException: Cannot copy from a TensorFlowLite tensor (Identity) with shape [1, 3, 6400, 85] to a Java object with shape [1, 20, 20, 425]

is thrown when trying to detect an object

Tflite.detectObjectOnFrame(
      bytesList: image.planes.map((plane) => plane.bytes).toList(),
      model: "YOLO",
      imageHeight: 640,
      imageWidth: 640,
      imageMean: 0,
      imageStd: 255.0,
      anchors: [
        10,
        13,
        16,
        30,
        33,
        23,
        30,
        61,
        62,
        45,
        59,
        119,
        116,
        90,
        156,
        198,
        373,
        326
      ],
    )

Does anyone has any idea about why this could be happening?