ashilkn пре 1 година
родитељ
комит
35a8a96898

+ 3 - 2
mobile/lib/services/machine_learning/semantic_search/frameworks/onnx/onnx_image_encoder.dart

@@ -15,9 +15,10 @@ class OnnxImageEncoder {
       ..setIntraOpNumThreads(1)
       ..setIntraOpNumThreads(1)
       ..setSessionGraphOptimizationLevel(GraphOptimizationLevel.ortEnableAll);
       ..setSessionGraphOptimizationLevel(GraphOptimizationLevel.ortEnableAll);
     try {
     try {
-      final bytes = await File(args["imageModelPath"]).readAsBytes();
-      final session = OrtSession.fromBuffer(bytes, sessionOptions);
+      final session =
+          OrtSession.fromFile(File(args["imageModelPath"]), sessionOptions);
       _logger.info('image model loaded');
       _logger.info('image model loaded');
+
       return session.address;
       return session.address;
     } catch (e, s) {
     } catch (e, s) {
       _logger.severe(e, s);
       _logger.severe(e, s);