From 351d5f85ac2cd5e418b83f0260fd8fc27d9207ce Mon Sep 17 00:00:00 2001
From: laurenspriem <github_uni.fishers@aleeas.com>
Date: Tue, 2 Jul 2024 14:27:37 +0530
Subject: [PATCH] [mob][photos] Minor change

---
 .../semantic_search/semantic_search_service.dart            | 5 +----
 mobile/lib/utils/ml_util.dart                               | 6 ++++--
 2 files changed, 5 insertions(+), 6 deletions(-)

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 be1bc08aab..0c717d9791 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
@@ -222,10 +222,7 @@ class SemanticSearchService {
     final embeddedFileIDs =
         await EmbeddingsDB.instance.getFileIDs(_currentModel);
 
-    uploadedFileIDs.removeWhere(
-      (id) => embeddedFileIDs.contains(id),
-    );
-    return uploadedFileIDs;
+    return uploadedFileIDs.difference(embeddedFileIDs).toList();
   }
 
   Future<void> clearQueue() async {
diff --git a/mobile/lib/utils/ml_util.dart b/mobile/lib/utils/ml_util.dart
index f440d6868f..c81a288dd2 100644
--- a/mobile/lib/utils/ml_util.dart
+++ b/mobile/lib/utils/ml_util.dart
@@ -14,8 +14,10 @@ final _logger = Logger("MlUtil");
 
 enum FileDataForML { thumbnailData, fileData }
 
-Future<List<int>> getIndexableFileIDs() async {
-  return FilesDB.instance.getOwnedFileIDs(Configuration.instance.getUserID()!);
+Future<Set<int>> getIndexableFileIDs() async {
+  final fileIDs = await FilesDB.instance
+      .getOwnedFileIDs(Configuration.instance.getUserID()!);
+  return fileIDs.toSet();
 }
 
 Future<String> getImagePathForML(