TerribleDev/ente
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[mob][photos] Abstract choice of onnx package

Browse Source
This commit is contained in:
laurenspriem
2024-07-25 17:54:18 +02:00
parent ea5eef0f2f
commit cdcc349157
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
mobile/lib/services/machine_learning/ml_model.dart
View File

@@ -16,10 +16,12 @@ abstract class MlModel {
String get modelName;
static final bool usePlatformPlugin = Platform.isAndroid;
bool get isInitialized =>
Platform.isAndroid ? isNativePluginInitialized : isFfiInitialized;
usePlatformPlugin ? isNativePluginInitialized : isFfiInitialized;
int get sessionAddress =>
Platform.isAndroid ? _nativePluginSessionIndex : _ffiSessionAddress;
usePlatformPlugin ? _nativePluginSessionIndex : _ffiSessionAddress;
// isInitialized is used to check if the model is loaded by the ffi based
// plugin
@@ -36,7 +38,7 @@ abstract class MlModel {
}
void storeSessionAddress(int address) {
if (Platform.isAndroid) {
if (usePlatformPlugin) {
_nativePluginSessionIndex = address;
isNativePluginInitialized = true;
} else {
@@ -53,7 +55,7 @@ abstract class MlModel {
String modelName,
String modelPath,
) async {
if (Platform.isAndroid) {
if (usePlatformPlugin) {
return await _loadModelWithEntePlugin(modelName, modelPath);
} else {
return await _loadModelWithFFI(modelName, modelPath);