From c8effce8fc9a22caf46a42d962b5f2d74d5811ba Mon Sep 17 00:00:00 2001
From: laurenspriem <github_uni.fishers@aleeas.com>
Date: Sat, 22 Jun 2024 00:33:26 +0530
Subject: [PATCH] [mob][photos] More cleanup

---
 .../face_ml/face_embedding/face_embedding_service.dart    | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/mobile/lib/services/machine_learning/face_ml/face_embedding/face_embedding_service.dart b/mobile/lib/services/machine_learning/face_ml/face_embedding/face_embedding_service.dart
index bee85cfa31..0e49cb8c4f 100644
--- a/mobile/lib/services/machine_learning/face_ml/face_embedding/face_embedding_service.dart
+++ b/mobile/lib/services/machine_learning/face_ml/face_embedding/face_embedding_service.dart
@@ -71,13 +71,11 @@ class FaceEmbeddingService {
       ..setIntraOpNumThreads(1)
       ..setSessionGraphOptimizationLevel(GraphOptimizationLevel.ortEnableAll);
     try {
-      // _logger.info('Loading face embedding model');
       final session =
           OrtSession.fromFile(File(args["modelPath"]), sessionOptions);
-      // _logger.info('Face embedding model loaded');
       return session.address;
-    } catch (e, _) {
-      // _logger.severe('Face embedding model not loaded', e, s);
+    } catch (e, s) {
+      _logger.severe('Face embedding model not loaded', e, s);
     }
     return -1;
   }
@@ -104,7 +102,7 @@ class FaceEmbeddingService {
       final int numberOfFaces = input.length ~/ (kInputSize * kInputSize * 3);
       final inputOrt = OrtValueTensor.createTensorWithDataList(
         input,
-        [numberOfFaces, kInputSize, kInputSize, 3],
+        [numberOfFaces, kInputSize, kInputSize, kNumChannels],
       );
       final inputs = {'img_inputs': inputOrt};
       final session = OrtSession.fromAddress(sessionAddress);