bundle moved to model

BundleURL.swift

@@ -0,0 +1,44 @@
+import CoreML
+
+@available(macOS 10.13, iOS 11.0, tvOS 11.0, watchOS 4.0, *)
+//model name : "FindFour.mlmodelc"
+struct BundleURL{
+    static func bundleUrl(forResource: String, withExtension: String, modelName: String) -> URL?{
+        guard let documentDirectory =
+            try? FileManager.default.url(for: .documentDirectory, in: .userDomainMask, appropriateFor:nil, create:false) else {
+                print("Directory could not be found")
+                return nil
+        }
+
+        let modelcFile = documentDirectory.appendingPathComponent(modelName)
+
+        if !FileManager.default.fileExists(atPath: modelcFile.path) {
+            guard let bundleUrl = Bundle(for: FindFour.self).url(forResource: "CardScan", withExtension: "bundle") else {
+                print("bundleURL could not be found")
+                return nil
+            }
+
+            guard let bundle = Bundle(url: bundleUrl) else {
+                print("bundle with bundleURL could not be found")
+                return nil
+            }
+
+            guard let modelUrl = bundle.url(forResource: forResource, withExtension: withExtension) else {
+                print("Could not find bundle named \" FindFour.bin\" ")
+                return nil
+            }
+
+            guard let compiledUrl = try? MLModel.compileModel(at: modelUrl) else {
+                print("Model could not compile")
+                return nil
+            }
+
+            guard let _ = try? FileManager.default.moveItem(at: compiledUrl, to: modelcFile) else {
+                print("Could not move to modelcFile")
+                return nil
+            }
+        }
+
+        return modelcFile
+    }
+}

CardScan/Classes/Models.swift

@@ -70,22 +70,7 @@ class FindFourOutput : MLFeatureProvider {
 @available(macOS 10.13, iOS 11.0, tvOS 11.0, watchOS 4.0, *)
 class FindFour {
     var model: MLModel
-
-    /// URL of model assuming it was installed in the same bundle as this class
-    class var urlOfModelInThisBundle : URL {
-        let bundleUrl = Bundle(for: FindFour.self).url(forResource: "CardScan", withExtension: "bundle")!
-        let bundle = Bundle(url: bundleUrl)!
-
-        let documentDirectory = try! FileManager.default.url(for: .documentDirectory, in: .userDomainMask, appropriateFor:nil, create:false)
-        let modelcFile = documentDirectory.appendingPathComponent("FindFour.mlmodelc")
-        if !FileManager.default.fileExists(atPath: modelcFile.path) {
-            let modelUrl = bundle.url(forResource: "FindFour", withExtension: "bin")!
-            let compiledUrl = try? MLModel.compileModel(at: modelUrl)
-            try! FileManager.default.moveItem(at: compiledUrl!, to: modelcFile)
-        }
-
-        return modelcFile
-    }
+
 
     /**
      Construct a model with explicit path to mlmodelc file
@@ -99,7 +84,7 @@ class FindFour {
 
     /// Construct a model that automatically loads the model from the app's bundle
     convenience init() {
-        try! self.init(contentsOf: type(of:self).urlOfModelInThisBundle)
+        try! self.init(contentsOf: BundleURL)
     }
 
     /**