From 56b4d3cd46c45d83952635dd5867283dfcf99e54 Mon Sep 17 00:00:00 2001 From: laurenspriem Date: Tue, 23 Jul 2024 11:48:14 +0200 Subject: [PATCH] [mob][photos] Rename for consistency --- .../semantic_search/clip/clip_text_encoder.dart | 5 +++-- .../semantic_search/semantic_search_service.dart | 2 +- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/mobile/lib/services/machine_learning/semantic_search/clip/clip_text_encoder.dart b/mobile/lib/services/machine_learning/semantic_search/clip/clip_text_encoder.dart index 6f74ab48c7..2c3e6b7e8a 100644 --- a/mobile/lib/services/machine_learning/semantic_search/clip/clip_text_encoder.dart +++ b/mobile/lib/services/machine_learning/semantic_search/clip/clip_text_encoder.dart @@ -30,11 +30,12 @@ class ClipTextEncoder extends MlModel { static final instance = ClipTextEncoder._privateConstructor(); factory ClipTextEncoder() => instance; - static Future> infer(Map args) async { + static Future> predict(Map args) async { final text = args["text"]; final address = args["address"] as int; final vocabPath = args["vocabPath"] as String; - final List tokenize = await ClipTextTokenizer.instance.tokenize(text, vocabPath); + final List tokenize = + await ClipTextTokenizer.instance.tokenize(text, vocabPath); final int32list = Int32List.fromList(tokenize); return _runFFIBasedPredict(int32list, address); } diff --git a/mobile/lib/services/machine_learning/semantic_search/semantic_search_service.dart b/mobile/lib/services/machine_learning/semantic_search/semantic_search_service.dart index ca45633ac1..caf0a879f0 100644 --- a/mobile/lib/services/machine_learning/semantic_search/semantic_search_service.dart +++ b/mobile/lib/services/machine_learning/semantic_search/semantic_search_service.dart @@ -302,7 +302,7 @@ class SemanticSearchService { final String tokenizerVocabPath = await RemoteAssetsService.instance.getAssetPath(remotePath); final textEmbedding = await _computer.compute( - ClipTextEncoder.infer, + ClipTextEncoder.predict, param: { "text": query, "address": clipAddress,