From f03cea7252ea33e4e090f9b8062bcafb9d0aa783 Mon Sep 17 00:00:00 2001 From: laurenspriem Date: Fri, 28 Jun 2024 15:56:27 +0530 Subject: [PATCH] [mob][photos] Inline --- .../semantic_search/frameworks/onnx/onnx_image_encoder.dart | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/mobile/lib/services/machine_learning/semantic_search/frameworks/onnx/onnx_image_encoder.dart b/mobile/lib/services/machine_learning/semantic_search/frameworks/onnx/onnx_image_encoder.dart index c750ad461d..460fcc8597 100644 --- a/mobile/lib/services/machine_learning/semantic_search/frameworks/onnx/onnx_image_encoder.dart +++ b/mobile/lib/services/machine_learning/semantic_search/frameworks/onnx/onnx_image_encoder.dart @@ -27,8 +27,6 @@ class OnnxImageEncoder { } Future> inferByImage(Map args) async { - final runOptions = OrtRunOptions(); - //Check the existence of imagePath locally final rgb = img.decodeImage(await File(args["imagePath"]).readAsBytes())!; final int imageWidth = rgb.width; @@ -105,6 +103,7 @@ class OnnxImageEncoder { OrtValueTensor.createTensorWithDataList(floatList, [1, 3, 224, 224]); final inputs = {'input': inputOrt}; final session = OrtSession.fromAddress(args["address"]); + final runOptions = OrtRunOptions(); final outputs = session.run(runOptions, inputs); final embedding = (outputs[0]?.value as List>)[0];