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(