Update ios/Classes/TflitePlugin.mm

* Don't shadow the `options` variable with a duplicate name in the CoreML code
* Ensure we clean up the coreml delegate correctly

Signed-off-by: Daniel Llewellyn <[email protected]>

Author: Daniel Llewellyn

ios/Classes/TflitePlugin.mm

@@ -119,6 +119,7 @@ - (void)handleMethodCall:(FlutterMethodCall*)call result:(FlutterResult)result {
 TfLiteInterpreter *interpreter = nullptr;
 TfLiteModel *model = nullptr;
 TfLiteDelegate *delegate = nullptr;
+bool isCoreMLDelegate = false;
 #else
 std::unique_ptr<tflite::FlatBufferModel> model;
 std::unique_ptr<tflite::Interpreter> interpreter;
@@ -161,12 +162,14 @@ static void LoadLabels(NSString* labels_path,
   }
   options = TfLiteInterpreterOptionsCreate();
   TfLiteInterpreterOptionsSetNumThreads(options, num_threads);
-  
+
   bool useGpuDelegate = [args[@"useGpuDelegate"] boolValue];
   bool useCoreMLDelegate = [args[@"useCoreMLDelegate"] boolValue];
   if (useCoreMLDelegate) {
-    TfLiteCoreMlDelegateOptions options = {};
-    delegate = TfLiteCoreMlDelegateCreate(&options);
+    delegate = TfLiteCoreMlDelegateCreate(nullptr);
+    if (delegate != NULL) {
+      isCoreMLDelegate = true;
+    }
   }
   if (delegate == NULL && useGpuDelegate) {
     delegate = TFLGpuDelegateCreate(nullptr);
@@ -1496,8 +1499,11 @@ void runPoseNetOnFrame(NSDictionary* args, FlutterResult result) {
 void close() {
 #ifdef TFLITE2
   interpreter = nullptr;
-  if (delegate != nullptr)
+  if (isCoreMLDelegate && delegate != nullptr) {
+    TfLiteCoreMlDelegateDelete(delegate);
+  } else if (delegate != nullptr) {
     TFLGpuDelegateDelete(delegate);
+  }
   delegate = nullptr;
 #else
   interpreter.release();