diff --git a/CapCollector.xcodeproj/project.xcworkspace/xcuserdata/imac.xcuserdatad/UserInterfaceState.xcuserstate b/CapCollector.xcodeproj/project.xcworkspace/xcuserdata/imac.xcuserdatad/UserInterfaceState.xcuserstate index c702d5e..4385934 100644 Binary files a/CapCollector.xcodeproj/project.xcworkspace/xcuserdata/imac.xcuserdatad/UserInterfaceState.xcuserstate and b/CapCollector.xcodeproj/project.xcworkspace/xcuserdata/imac.xcuserdatad/UserInterfaceState.xcuserstate differ diff --git a/CapCollector/TableView.swift b/CapCollector/TableView.swift index 7725f05..91c59e5 100644 --- a/CapCollector/TableView.swift +++ b/CapCollector/TableView.swift @@ -495,8 +495,8 @@ class TableView: UITableViewController { updateNavigationItemTitleView() } - private func loadClassifier() { - guard classifier == nil else { + private func loadClassifier(reload: Bool = false) { + guard classifier == nil || reload else { return } guard let model = app.storage.recognitionModel else { @@ -691,7 +691,7 @@ class TableView: UITableViewController { self.log("Failed to download classifier") return } - loadClassifier() + loadClassifier(reload: true) self.log("Classifier was downloaded.") guard let image = accessory!.currentImage else { classifyDummyImage()