diff --git a/mobile/lib/services/machine_learning/ml_computer.dart b/mobile/lib/services/machine_learning/ml_computer.dart index 9c4fabdd4f..753725abc0 100644 --- a/mobile/lib/services/machine_learning/ml_computer.dart +++ b/mobile/lib/services/machine_learning/ml_computer.dart @@ -21,7 +21,7 @@ enum MLComputerOperation { } class MLComputer { - final _logger = Logger('MLComputerIsolate'); + final _logger = Logger('MLComputer'); final _initLock = Lock(); final _functionLock = Lock(); @@ -99,7 +99,7 @@ class MLComputer { break; case MLComputerOperation.runClipText: final textEmbedding = await ClipTextEncoder.predict(args); - sendPort.send(List.from(textEmbedding, growable: false)); + sendPort.send(List.from(textEmbedding, growable: false)); break; } } catch (e, stackTrace) { @@ -158,8 +158,8 @@ class MLComputer { } Future> runClipText(String query) async { + await _ensureLoadedClipTextModel(); try { - await _ensureLoadedClipTextModel(); final int clipAddress = ClipTextEncoder.instance.sessionAddress; final textEmbedding = await _runInIsolate( ( diff --git a/mobile/lib/services/machine_learning/ml_indexing_isolate.dart b/mobile/lib/services/machine_learning/ml_indexing_isolate.dart index 481dc7915c..7636135289 100644 --- a/mobile/lib/services/machine_learning/ml_indexing_isolate.dart +++ b/mobile/lib/services/machine_learning/ml_indexing_isolate.dart @@ -99,7 +99,7 @@ class MLIndexingIsolate { ); addresses.add(address); } - sendPort.send(List.from(addresses, growable: false)); + sendPort.send(List.from(addresses, growable: false)); break; case MLIndexingOperation.releaseModels: final modelNames = args['modelNames'] as List;