Use CPUOnly when using VisionKit VNCoreMLRequest on Simulator

lovoo · Aug 18, 2024 · 9f290be · 9f290be
1 parent 5beac79
commit 9f290be
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/NSFWDetector/Classes/NSFWDetector.swift b/NSFWDetector/Classes/NSFWDetector.swift
@@ -77,6 +77,18 @@ private extension NSFWDetector {
             completion(.success(nsfwConfidence: observation.confidence))
         })
 
+        /**
+         * `@Required`:
+         * Set to true on Simulator or VisionKit will attempt to use GPU and fail
+         *
+         * `@Important`:
+         * Running on Simulator results in `significantly reduced accuracy`.
+         * Run on physical device for acurate results
+         */
+        #if targetEnvironment(simulator)
+        request.usesCPUOnly = true
+        #endif
+
         /// Start the actual detection
         do {
             try requestHandler.perform([request])