From e3908f3f2c1b4175d743498f07a645fa10c2eb28 Mon Sep 17 00:00:00 2001 From: laurenspriem Date: Thu, 25 Jul 2024 17:55:49 +0200 Subject: [PATCH] [mob][photos] Minor change --- .../services/machine_learning/ml_model.dart | 22 +++++++++---------- 1 file changed, 11 insertions(+), 11 deletions(-) diff --git a/mobile/lib/services/machine_learning/ml_model.dart b/mobile/lib/services/machine_learning/ml_model.dart index e2097beff4..1770bdfa4d 100644 --- a/mobile/lib/services/machine_learning/ml_model.dart +++ b/mobile/lib/services/machine_learning/ml_model.dart @@ -16,19 +16,19 @@ abstract class MlModel { String get modelName; - static final bool usePlatformPlugin = Platform.isAndroid; + static final bool _usePlatformPlugin = Platform.isAndroid; bool get isInitialized => - usePlatformPlugin ? isNativePluginInitialized : isFfiInitialized; + _usePlatformPlugin ? _isNativePluginInitialized : _isFfiInitialized; int get sessionAddress => - usePlatformPlugin ? _nativePluginSessionIndex : _ffiSessionAddress; + _usePlatformPlugin ? _nativePluginSessionIndex : _ffiSessionAddress; // isInitialized is used to check if the model is loaded by the ffi based // plugin - bool isFfiInitialized = false; + bool _isFfiInitialized = false; int _ffiSessionAddress = -1; - bool isNativePluginInitialized = false; + bool _isNativePluginInitialized = false; int _nativePluginSessionIndex = -1; Future<(String, String)> getModelNameAndPath() async { @@ -38,12 +38,12 @@ abstract class MlModel { } void storeSessionAddress(int address) { - if (usePlatformPlugin) { + if (_usePlatformPlugin) { _nativePluginSessionIndex = address; - isNativePluginInitialized = true; + _isNativePluginInitialized = true; } else { _ffiSessionAddress = address; - isFfiInitialized = true; + _isFfiInitialized = true; } } @@ -55,7 +55,7 @@ abstract class MlModel { String modelName, String modelPath, ) async { - if (usePlatformPlugin) { + if (_usePlatformPlugin) { return await _loadModelWithEntePlugin(modelName, modelPath); } else { return await _loadModelWithFFI(modelName, modelPath); @@ -107,10 +107,10 @@ abstract class MlModel { // TODO: add release method for native plugin Future release() async { - if (isFfiInitialized) { + if (_isFfiInitialized) { await _releaseModel({'address': _ffiSessionAddress}); await ONNXEnv.instance.releaseONNX(modelName); - isFfiInitialized = false; + _isFfiInitialized = false; _ffiSessionAddress = 0; } }