0ct0pu5/ente
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Read data asynchronously

Browse source
This commit is contained in:
vishnukvmd 2024-01-24 22:18:21 +05:30
parent 009bda8ca8
commit 33bb3e9b5e
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
lib/services/semantic_search/frameworks/onnx/onnx_image_encoder.dart
View file

@ -15,7 +15,7 @@ class OnnxImageEncoder {
..setIntraOpNumThreads(1)
..setSessionGraphOptimizationLevel(GraphOptimizationLevel.ortEnableAll);
try {
final bytes = File(args["imageModelPath"]).readAsBytesSync();
final bytes = await File(args["imageModelPath"]).readAsBytes();
final session = OrtSession.fromBuffer(bytes, sessionOptions);
_logger.info('image model loaded');
return session.address;
@ -25,10 +25,10 @@ class OnnxImageEncoder {
return -1;
}
List<double> inferByImage(Map args) {
Future<List<double>> inferByImage(Map args) async {
final runOptions = OrtRunOptions();
//Check the existence of imagePath locally
final rgb = img.decodeImage(File(args["imagePath"]).readAsBytesSync())!;
final rgb = img.decodeImage(await File(args["imagePath"]).readAsBytes())!;
final int nx = rgb.width;
final int ny = rgb.height;