I have a issue in runSegmentationOnImage

[matifdeveloper]

I am using this model https://tfhub.dev/sayannath/lite-model/zero-dce/1

My Code: Future runZeroDceModel(File image) async { try { // Resize the image

  final imageBytes = await image.readAsBytes();
  final decodedImage = img.decodeImage(imageBytes.toList());

  final resizedImage = img.copyResize(decodedImage!, width: 400, height: 600);
  final resizedFile = File(image.path);
  await resizedFile.writeAsBytes(img.encodeJpg(resizedImage));

  // Run ZeroDce model on the resized image
  var recognitions = await Tflite.runSegmentationOnImage(
    path: resizedFile.path,  // Use the resized image path
    //imageMean: 0.0,
    //imageStd: 255.0,
    outputType: "png",
    asynch: true,
  );

  if (recognitions != null) {
    log('recognitions: $recognitions');

  } else {
    log('recognitions: error');
  }

  log('result: $recognitions');
} catch (e) {
  log('Error: $e');
}

}

Error: E/AndroidRuntime(11561): FATAL EXCEPTION: main E/AndroidRuntime(11561): Process: com.example.fyp_deaf, PID: 11561 E/AndroidRuntime(11561): java.lang.ArrayIndexOutOfBoundsException: length=2; index=2 E/AndroidRuntime(11561): at sq.flutter.flutter_tflite.TflitePlugin.fetchArgmax(TflitePlugin.java:1168) E/AndroidRuntime(11561): at sq.flutter.flutter_tflite.TflitePlugin$RunSegmentationOnImage.onRunTfliteDone(TflitePlugin.java:1071) E/AndroidRuntime(11561): at sq.flutter.flutter_tflite.TflitePlugin$TfliteTask.onPostExecute(TflitePlugin.java:513) E/AndroidRuntime(11561): at sq.flutter.flutter_tflite.TflitePlugin$TfliteTask.onPostExecute(TflitePlugin.java:481) E/AndroidRuntime(11561): at android.os.AsyncTask.finish(AsyncTask.java:771) E/AndroidRuntime(11561): at android.os.AsyncTask.-$$Nest$mfinish(Unknown Source:0) E/AndroidRuntime(11561): at android.os.AsyncTask$InternalHandler.handleMessage(AsyncTask.java:788) E/AndroidRuntime(11561): at android.os.Handler.dispatchMessage(Handler.java:106) E/AndroidRuntime(11561): at android.os.Looper.loopOnce(Looper.java:201) E/AndroidRuntime(11561): at android.os.Looper.loop(Looper.java:288) E/AndroidRuntime(11561): at android.app.ActivityThread.main(ActivityThread.java:7872) E/AndroidRuntime(11561): at java.lang.reflect.Method.invoke(Native Method) E/AndroidRuntime(11561): at com.android.internal.os.RuntimeInit$MethodAndArgsCaller.run(RuntimeInit.java:548) E/AndroidRuntime(11561): at com.android.internal.os.ZygoteInit.main(ZygoteInit.java:936)