diff --git a/.github/workflows/format.yml b/.github/workflows/format.yml
index f988ad3..5b39b9b 100644
--- a/.github/workflows/format.yml
+++ b/.github/workflows/format.yml
@@ -13,7 +13,7 @@ on:
 
 jobs:
   format:
-    runs-on: ubuntu-latest
+    runs-on: macos-latest
     steps:
       - name: Run Ultralytics Formatting
         uses: ultralytics/actions@main
@@ -23,6 +23,7 @@ jobs:
           python: true # format Python code and docstrings
           markdown: true # format Markdown
           prettier: true # format YAML
+          swift: true # format Swift
           spelling: false # check spelling
           links: false # check broken links
           summary: true # print PR summary with GPT4 (requires 'openai_api_key' or 'openai_azure_api_key' and 'openai_azure_endpoint')
diff --git a/YOLO.xcodeproj/project.pbxproj b/YOLO.xcodeproj/project.pbxproj
index ddcf965..7176d44 100644
--- a/YOLO.xcodeproj/project.pbxproj
+++ b/YOLO.xcodeproj/project.pbxproj
@@ -13,14 +13,12 @@
 		636EFCAF21E62DD300DE43BC /* VideoCapture.swift in Sources */ = {isa = PBXBuildFile; fileRef = 636EFCA221E62DD300DE43BC /* VideoCapture.swift */; };
 		636EFCB321E62DD300DE43BC /* AppDelegate.swift in Sources */ = {isa = PBXBuildFile; fileRef = 636EFCA721E62DD300DE43BC /* AppDelegate.swift */; };
 		636EFCB921E62E3900DE43BC /* Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 636EFCB821E62E3900DE43BC /* Assets.xcassets */; };
-		6381D2182B7817C200ABA4E8 /* yolov8l.mlpackage in Sources */ = {isa = PBXBuildFile; fileRef = 6381D2132B7817C200ABA4E8 /* yolov8l.mlpackage */; };
-		6381D2192B7817C200ABA4E8 /* yolov8x.mlpackage in Sources */ = {isa = PBXBuildFile; fileRef = 6381D2142B7817C200ABA4E8 /* yolov8x.mlpackage */; };
-		6381D21A2B7817C200ABA4E8 /* yolov8s.mlpackage in Sources */ = {isa = PBXBuildFile; fileRef = 6381D2152B7817C200ABA4E8 /* yolov8s.mlpackage */; };
-		6381D21B2B7817C200ABA4E8 /* yolov8m.mlpackage in Sources */ = {isa = PBXBuildFile; fileRef = 6381D2162B7817C200ABA4E8 /* yolov8m.mlpackage */; };
-		6381D21C2B7817C200ABA4E8 /* yolov8n.mlpackage in Sources */ = {isa = PBXBuildFile; fileRef = 6381D2172B7817C200ABA4E8 /* yolov8n.mlpackage */; };
 		63CF371F2514455300E2DEA1 /* LaunchScreen.storyboard in Resources */ = {isa = PBXBuildFile; fileRef = 6323C44D22186177008AE681 /* LaunchScreen.storyboard */; };
 		63CF37202514455300E2DEA1 /* Main.storyboard in Resources */ = {isa = PBXBuildFile; fileRef = 6323C44F22186177008AE681 /* Main.storyboard */; };
 		63CF37212514455300E2DEA1 /* ultralytics_yolo_logotype.png in Resources */ = {isa = PBXBuildFile; fileRef = 6323C45122186177008AE681 /* ultralytics_yolo_logotype.png */; };
+		73FD886D2C5931B700CFF647 /* Models.swift in Sources */ = {isa = PBXBuildFile; fileRef = 73FD886B2C5931B700CFF647 /* Models.swift */; };
+		73FD886E2C5931B700CFF647 /* ModelChacheManager.swift in Sources */ = {isa = PBXBuildFile; fileRef = 73FD886C2C5931B700CFF647 /* ModelChacheManager.swift */; };
+		73FD88752C59411600CFF647 /* ZIPFoundation in Frameworks */ = {isa = PBXBuildFile; productRef = 73FD88742C59411600CFF647 /* ZIPFoundation */; };
 		8EDAA33950796844333D60A7 /* BoundingBoxView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 8EDAA633C1F2B50286D16008 /* BoundingBoxView.swift */; };
 /* End PBXBuildFile section */
 
@@ -35,12 +33,9 @@
 		636EFCA221E62DD300DE43BC /* VideoCapture.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = VideoCapture.swift; sourceTree = "<group>"; };
 		636EFCA721E62DD300DE43BC /* AppDelegate.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = AppDelegate.swift; sourceTree = "<group>"; };
 		636EFCB821E62E3900DE43BC /* Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = Assets.xcassets; sourceTree = "<group>"; };
-		6381D2132B7817C200ABA4E8 /* yolov8l.mlpackage */ = {isa = PBXFileReference; lastKnownFileType = folder.mlpackage; path = yolov8l.mlpackage; sourceTree = "<group>"; };
-		6381D2142B7817C200ABA4E8 /* yolov8x.mlpackage */ = {isa = PBXFileReference; lastKnownFileType = folder.mlpackage; path = yolov8x.mlpackage; sourceTree = "<group>"; };
-		6381D2152B7817C200ABA4E8 /* yolov8s.mlpackage */ = {isa = PBXFileReference; lastKnownFileType = folder.mlpackage; path = yolov8s.mlpackage; sourceTree = "<group>"; };
-		6381D2162B7817C200ABA4E8 /* yolov8m.mlpackage */ = {isa = PBXFileReference; lastKnownFileType = folder.mlpackage; path = yolov8m.mlpackage; sourceTree = "<group>"; };
-		6381D2172B7817C200ABA4E8 /* yolov8n.mlpackage */ = {isa = PBXFileReference; lastKnownFileType = folder.mlpackage; path = yolov8n.mlpackage; sourceTree = "<group>"; };
 		63B8B0A821E62A890026FBC3 /* .gitignore */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = text; path = .gitignore; sourceTree = "<group>"; };
+		73FD886B2C5931B700CFF647 /* Models.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = Models.swift; sourceTree = "<group>"; };
+		73FD886C2C5931B700CFF647 /* ModelChacheManager.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = ModelChacheManager.swift; sourceTree = "<group>"; };
 		7BCB411721C3096100BFC4D0 /* YOLO.app */ = {isa = PBXFileReference; explicitFileType = wrapper.application; includeInIndex = 0; path = YOLO.app; sourceTree = BUILT_PRODUCTS_DIR; };
 		8EDAA633C1F2B50286D16008 /* BoundingBoxView.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = BoundingBoxView.swift; sourceTree = "<group>"; };
 		8EDAAA4507D2D23D7FAB827F /* README.md */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = net.daringfireball.markdown; path = README.md; sourceTree = "<group>"; };
@@ -51,6 +46,7 @@
 			isa = PBXFrameworksBuildPhase;
 			buildActionMask = 2147483647;
 			files = (
+				73FD88752C59411600CFF647 /* ZIPFoundation in Frameworks */,
 			);
 			runOnlyForDeploymentPostprocessing = 0;
 		};
@@ -62,6 +58,8 @@
 			children = (
 				636166E9251443B20054FA7E /* ThresholdProvider.swift */,
 				8EDAA633C1F2B50286D16008 /* BoundingBoxView.swift */,
+				73FD886C2C5931B700CFF647 /* ModelChacheManager.swift */,
+				73FD886B2C5931B700CFF647 /* Models.swift */,
 			);
 			path = Utilities;
 			sourceTree = "<group>";
@@ -87,11 +85,6 @@
 		63A946D8271800E20001C3ED /* Models */ = {
 			isa = PBXGroup;
 			children = (
-				6381D2132B7817C200ABA4E8 /* yolov8l.mlpackage */,
-				6381D2162B7817C200ABA4E8 /* yolov8m.mlpackage */,
-				6381D2172B7817C200ABA4E8 /* yolov8n.mlpackage */,
-				6381D2152B7817C200ABA4E8 /* yolov8s.mlpackage */,
-				6381D2142B7817C200ABA4E8 /* yolov8x.mlpackage */,
 			);
 			path = Models;
 			sourceTree = "<group>";
@@ -131,6 +124,9 @@
 			dependencies = (
 			);
 			name = YOLO;
+			packageProductDependencies = (
+				73FD88742C59411600CFF647 /* ZIPFoundation */,
+			);
 			productName = YOLO;
 			productReference = 7BCB411721C3096100BFC4D0 /* YOLO.app */;
 			productType = "com.apple.product-type.application";
@@ -160,6 +156,9 @@
 				Base,
 			);
 			mainGroup = 7BCB410E21C3096100BFC4D0;
+			packageReferences = (
+				73FD88732C59411600CFF647 /* XCRemoteSwiftPackageReference "ZIPFoundation" */,
+			);
 			productRefGroup = 7BCB411821C3096100BFC4D0 /* Products */;
 			projectDirPath = "";
 			projectRoot = "";
@@ -210,13 +209,10 @@
 			isa = PBXSourcesBuildPhase;
 			buildActionMask = 2147483647;
 			files = (
-				6381D21B2B7817C200ABA4E8 /* yolov8m.mlpackage in Sources */,
-				6381D21C2B7817C200ABA4E8 /* yolov8n.mlpackage in Sources */,
+				73FD886D2C5931B700CFF647 /* Models.swift in Sources */,
 				636EFCAF21E62DD300DE43BC /* VideoCapture.swift in Sources */,
 				636166EA251443B20054FA7E /* ThresholdProvider.swift in Sources */,
-				6381D2182B7817C200ABA4E8 /* yolov8l.mlpackage in Sources */,
-				6381D21A2B7817C200ABA4E8 /* yolov8s.mlpackage in Sources */,
-				6381D2192B7817C200ABA4E8 /* yolov8x.mlpackage in Sources */,
+				73FD886E2C5931B700CFF647 /* ModelChacheManager.swift in Sources */,
 				636EFCB321E62DD300DE43BC /* AppDelegate.swift in Sources */,
 				636EFCAA21E62DD300DE43BC /* ViewController.swift in Sources */,
 				8EDAA33950796844333D60A7 /* BoundingBoxView.swift in Sources */,
@@ -351,7 +347,7 @@
 				ASSETCATALOG_COMPILER_INCLUDE_ALL_APPICON_ASSETS = NO;
 				CODE_SIGN_STYLE = Automatic;
 				CURRENT_PROJECT_VERSION = 0;
-				DEVELOPMENT_TEAM = 3MR4P6CL3X;
+				DEVELOPMENT_TEAM = MFN25KNUGJ;
 				INFOPLIST_FILE = YOLO/Info.plist;
 				INFOPLIST_KEY_CFBundleDisplayName = "Ultralytics YOLO";
 				INFOPLIST_KEY_LSApplicationCategoryType = "public.app-category.developer-tools";
@@ -379,7 +375,7 @@
 				ASSETCATALOG_COMPILER_INCLUDE_ALL_APPICON_ASSETS = NO;
 				CODE_SIGN_STYLE = Automatic;
 				CURRENT_PROJECT_VERSION = 0;
-				DEVELOPMENT_TEAM = 3MR4P6CL3X;
+				DEVELOPMENT_TEAM = MFN25KNUGJ;
 				INFOPLIST_FILE = YOLO/Info.plist;
 				INFOPLIST_KEY_CFBundleDisplayName = "Ultralytics YOLO";
 				INFOPLIST_KEY_LSApplicationCategoryType = "public.app-category.developer-tools";
@@ -422,6 +418,25 @@
 			defaultConfigurationName = Release;
 		};
 /* End XCConfigurationList section */
+
+/* Begin XCRemoteSwiftPackageReference section */
+		73FD88732C59411600CFF647 /* XCRemoteSwiftPackageReference "ZIPFoundation" */ = {
+			isa = XCRemoteSwiftPackageReference;
+			repositoryURL = "https://github.com/weichsel/ZIPFoundation.git";
+			requirement = {
+				kind = upToNextMajorVersion;
+				minimumVersion = 0.9.19;
+			};
+		};
+/* End XCRemoteSwiftPackageReference section */
+
+/* Begin XCSwiftPackageProductDependency section */
+		73FD88742C59411600CFF647 /* ZIPFoundation */ = {
+			isa = XCSwiftPackageProductDependency;
+			package = 73FD88732C59411600CFF647 /* XCRemoteSwiftPackageReference "ZIPFoundation" */;
+			productName = ZIPFoundation;
+		};
+/* End XCSwiftPackageProductDependency section */
 	};
 	rootObject = 7BCB410F21C3096100BFC4D0 /* Project object */;
 }
diff --git a/YOLO.xcodeproj/xcshareddata/xcschemes/YOLO.xcscheme b/YOLO.xcodeproj/xcshareddata/xcschemes/YOLO.xcscheme
new file mode 100644
index 0000000..3bb677d
--- /dev/null
+++ b/YOLO.xcodeproj/xcshareddata/xcschemes/YOLO.xcscheme
@@ -0,0 +1,78 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<Scheme
+   LastUpgradeVersion = "1540"
+   version = "1.7">
+   <BuildAction
+      parallelizeBuildables = "YES"
+      buildImplicitDependencies = "YES"
+      buildArchitectures = "Automatic">
+      <BuildActionEntries>
+         <BuildActionEntry
+            buildForTesting = "YES"
+            buildForRunning = "YES"
+            buildForProfiling = "YES"
+            buildForArchiving = "YES"
+            buildForAnalyzing = "YES">
+            <BuildableReference
+               BuildableIdentifier = "primary"
+               BlueprintIdentifier = "7BCB411621C3096100BFC4D0"
+               BuildableName = "YOLO.app"
+               BlueprintName = "YOLO"
+               ReferencedContainer = "container:YOLO.xcodeproj">
+            </BuildableReference>
+         </BuildActionEntry>
+      </BuildActionEntries>
+   </BuildAction>
+   <TestAction
+      buildConfiguration = "Debug"
+      selectedDebuggerIdentifier = "Xcode.DebuggerFoundation.Debugger.LLDB"
+      selectedLauncherIdentifier = "Xcode.DebuggerFoundation.Launcher.LLDB"
+      shouldUseLaunchSchemeArgsEnv = "YES"
+      shouldAutocreateTestPlan = "YES">
+   </TestAction>
+   <LaunchAction
+      buildConfiguration = "Debug"
+      selectedDebuggerIdentifier = "Xcode.DebuggerFoundation.Debugger.LLDB"
+      selectedLauncherIdentifier = "Xcode.DebuggerFoundation.Launcher.LLDB"
+      launchStyle = "0"
+      useCustomWorkingDirectory = "NO"
+      ignoresPersistentStateOnLaunch = "NO"
+      debugDocumentVersioning = "YES"
+      debugServiceExtension = "internal"
+      allowLocationSimulation = "YES">
+      <BuildableProductRunnable
+         runnableDebuggingMode = "0">
+         <BuildableReference
+            BuildableIdentifier = "primary"
+            BlueprintIdentifier = "7BCB411621C3096100BFC4D0"
+            BuildableName = "YOLO.app"
+            BlueprintName = "YOLO"
+            ReferencedContainer = "container:YOLO.xcodeproj">
+         </BuildableReference>
+      </BuildableProductRunnable>
+   </LaunchAction>
+   <ProfileAction
+      buildConfiguration = "Release"
+      shouldUseLaunchSchemeArgsEnv = "YES"
+      savedToolIdentifier = ""
+      useCustomWorkingDirectory = "NO"
+      debugDocumentVersioning = "YES">
+      <BuildableProductRunnable
+         runnableDebuggingMode = "0">
+         <BuildableReference
+            BuildableIdentifier = "primary"
+            BlueprintIdentifier = "7BCB411621C3096100BFC4D0"
+            BuildableName = "YOLO.app"
+            BlueprintName = "YOLO"
+            ReferencedContainer = "container:YOLO.xcodeproj">
+         </BuildableReference>
+      </BuildableProductRunnable>
+   </ProfileAction>
+   <AnalyzeAction
+      buildConfiguration = "Debug">
+   </AnalyzeAction>
+   <ArchiveAction
+      buildConfiguration = "Release"
+      revealArchiveInOrganizer = "YES">
+   </ArchiveAction>
+</Scheme>
diff --git a/YOLO/AppDelegate.swift b/YOLO/AppDelegate.swift
index fe2f900..d9b8711 100644
--- a/YOLO/AppDelegate.swift
+++ b/YOLO/AppDelegate.swift
@@ -17,49 +17,53 @@ import UIKit
 /// The main application delegate, handling global app behavior and configuration.
 @UIApplicationMain
 class AppDelegate: UIResponder, UIApplicationDelegate {
-    var window: UIWindow?
+  var window: UIWindow?
 
-    /// Called when the app finishes launching, used here to set global app settings.
-    func application(_ application: UIApplication, didFinishLaunchingWithOptions launchOptions: [UIApplication.LaunchOptionsKey: Any]?) -> Bool {
-        // Disable screen dimming and auto-lock to keep the app active during long operations.
-        UIApplication.shared.isIdleTimerDisabled = true
+  /// Called when the app finishes launching, used here to set global app settings.
+  func application(
+    _ application: UIApplication,
+    didFinishLaunchingWithOptions launchOptions: [UIApplication.LaunchOptionsKey: Any]?
+  ) -> Bool {
+    // Disable screen dimming and auto-lock to keep the app active during long operations.
+    UIApplication.shared.isIdleTimerDisabled = true
 
-        // Enable battery monitoring to allow the app to adapt its behavior based on battery level.
-        UIDevice.current.isBatteryMonitoringEnabled = true
+    // Enable battery monitoring to allow the app to adapt its behavior based on battery level.
+    UIDevice.current.isBatteryMonitoringEnabled = true
 
-        // Store the app version and build version in UserDefaults for easy access elsewhere in the app.
-        if let appVersion = Bundle.main.infoDictionary?["CFBundleShortVersionString"] as? String,
-           let buildVersion = Bundle.main.infoDictionary?["CFBundleVersion"] as? String {
-            UserDefaults.standard.set("\(appVersion) (\(buildVersion))", forKey: "app_version")
-        }
+    // Store the app version and build version in UserDefaults for easy access elsewhere in the app.
+    if let appVersion = Bundle.main.infoDictionary?["CFBundleShortVersionString"] as? String,
+      let buildVersion = Bundle.main.infoDictionary?["CFBundleVersion"] as? String
+    {
+      UserDefaults.standard.set("\(appVersion) (\(buildVersion))", forKey: "app_version")
+    }
 
-        // Store the device's UUID in UserDefaults for identification purposes.
-        if let uuid = UIDevice.current.identifierForVendor?.uuidString {
-            UserDefaults.standard.set(uuid, forKey: "uuid")
-        }
+    // Store the device's UUID in UserDefaults for identification purposes.
+    if let uuid = UIDevice.current.identifierForVendor?.uuidString {
+      UserDefaults.standard.set(uuid, forKey: "uuid")
+    }
 
-        // Ensure UserDefaults changes are immediately saved.
-        UserDefaults.standard.synchronize()
+    // Ensure UserDefaults changes are immediately saved.
+    UserDefaults.standard.synchronize()
 
-        return true
-    }
+    return true
+  }
 }
 
 /// Extension to CALayer to add functionality for generating screenshots of any layer.
 extension CALayer {
-    var screenShot: UIImage? {
-        // Begin a new image context, using the device's screen scale to ensure high-resolution output.
-        UIGraphicsBeginImageContextWithOptions(frame.size, false, UIScreen.main.scale)
-        defer {
-            UIGraphicsEndImageContext()
-        }  // Ensure the image context is cleaned up correctly.
+  var screenShot: UIImage? {
+    // Begin a new image context, using the device's screen scale to ensure high-resolution output.
+    UIGraphicsBeginImageContextWithOptions(frame.size, false, UIScreen.main.scale)
+    defer {
+      UIGraphicsEndImageContext()
+    }  // Ensure the image context is cleaned up correctly.
 
-        if let context = UIGraphicsGetCurrentContext() {
-            // Render the layer into the current context.
-            render(in: context)
-            // Attempt to generate an image from the current context.
-            return UIGraphicsGetImageFromCurrentImageContext()
-        }
-        return nil  // Return nil if the operation fails.
+    if let context = UIGraphicsGetCurrentContext() {
+      // Render the layer into the current context.
+      render(in: context)
+      // Attempt to generate an image from the current context.
+      return UIGraphicsGetImageFromCurrentImageContext()
     }
+    return nil  // Return nil if the operation fails.
+  }
 }
diff --git a/YOLO/Info.plist b/YOLO/Info.plist
index c36dbc0..5489cfb 100644
--- a/YOLO/Info.plist
+++ b/YOLO/Info.plist
@@ -21,7 +21,7 @@
 	<key>CFBundleShortVersionString</key>
 	<string>$(MARKETING_VERSION)</string>
 	<key>CFBundleVersion</key>
-	<string>24</string>
+	<string>78</string>
 	<key>ITSAppUsesNonExemptEncryption</key>
 	<false/>
 	<key>LSRequiresIPhoneOS</key>
diff --git a/YOLO/Main.storyboard b/YOLO/Main.storyboard
index 048e9f6..24a9283 100644
--- a/YOLO/Main.storyboard
+++ b/YOLO/Main.storyboard
@@ -1,9 +1,9 @@
 <?xml version="1.0" encoding="UTF-8"?>
-<document type="com.apple.InterfaceBuilder3.CocoaTouch.Storyboard.XIB" version="3.0" toolsVersion="22505" targetRuntime="iOS.CocoaTouch" propertyAccessControl="none" useAutolayout="YES" useTraitCollections="YES" useSafeAreas="YES" colorMatched="YES" initialViewController="BYZ-38-t0r">
+<document type="com.apple.InterfaceBuilder3.CocoaTouch.Storyboard.XIB" version="3.0" toolsVersion="32700.99.1234" targetRuntime="iOS.CocoaTouch" propertyAccessControl="none" useAutolayout="YES" useTraitCollections="YES" useSafeAreas="YES" colorMatched="YES" initialViewController="BYZ-38-t0r">
     <device id="retina6_12" orientation="portrait" appearance="light"/>
     <dependencies>
         <deployment identifier="iOS"/>
-        <plugIn identifier="com.apple.InterfaceBuilder.IBCocoaTouchPlugin" version="22504"/>
+        <plugIn identifier="com.apple.InterfaceBuilder.IBCocoaTouchPlugin" version="22685"/>
         <capability name="Safe area layout guides" minToolsVersion="9.0"/>
         <capability name="System colors in document resources" minToolsVersion="11.0"/>
         <capability name="documents saved in the Xcode 8 format" minToolsVersion="8.0"/>
@@ -34,16 +34,14 @@
                                 <color key="textColor" systemColor="groupTableViewBackgroundColor"/>
                                 <nil key="highlightedColor"/>
                             </label>
-                            <label opaque="NO" userInteractionEnabled="NO" contentMode="left" horizontalHuggingPriority="251" verticalHuggingPriority="251" fixedFrame="YES" text="Label" textAlignment="center" lineBreakMode="tailTruncation" baselineAdjustment="alignBaselines" adjustsFontForContentSizeCategory="YES" adjustsFontSizeToFit="NO" translatesAutoresizingMaskIntoConstraints="NO" id="Hif-Vv-5nn">
-                                <rect key="frame" x="13" y="59" width="367" height="32"/>
-                                <autoresizingMask key="autoresizingMask" flexibleMinX="YES" flexibleMaxX="YES" flexibleMinY="YES" flexibleMaxY="YES"/>
+                            <label opaque="NO" userInteractionEnabled="NO" contentMode="left" horizontalHuggingPriority="251" verticalHuggingPriority="251" text="Label" textAlignment="center" lineBreakMode="tailTruncation" baselineAdjustment="alignBaselines" adjustsFontForContentSizeCategory="YES" adjustsFontSizeToFit="NO" translatesAutoresizingMaskIntoConstraints="NO" id="Hif-Vv-5nn">
+                                <rect key="frame" x="163.66666666666666" y="59" width="66" height="34"/>
                                 <fontDescription key="fontDescription" style="UICTFontTextStyleTitle1"/>
                                 <color key="textColor" systemColor="groupTableViewBackgroundColor"/>
                                 <nil key="highlightedColor"/>
                             </label>
-                            <segmentedControl opaque="NO" contentMode="scaleToFill" fixedFrame="YES" contentHorizontalAlignment="center" contentVerticalAlignment="center" apportionsSegmentWidthsByContent="YES" segmentControlStyle="plain" selectedSegmentIndex="2" springLoaded="YES" translatesAutoresizingMaskIntoConstraints="NO" id="9cm-Nq-b3N">
-                                <rect key="frame" x="19" y="98" width="355" height="32"/>
-                                <autoresizingMask key="autoresizingMask" flexibleMinX="YES" flexibleMaxX="YES" flexibleMinY="YES" flexibleMaxY="YES"/>
+                            <segmentedControl opaque="NO" contentMode="scaleToFill" contentHorizontalAlignment="center" contentVerticalAlignment="center" apportionsSegmentWidthsByContent="YES" segmentControlStyle="plain" selectedSegmentIndex="2" springLoaded="YES" translatesAutoresizingMaskIntoConstraints="NO" id="9cm-Nq-b3N">
+                                <rect key="frame" x="19" y="100" width="355" height="32"/>
                                 <segments>
                                     <segment title="YOLOv8n" width="69"/>
                                     <segment title="YOLOv8s" width="69"/>
@@ -182,26 +180,137 @@
                                 </items>
                                 <color key="tintColor" white="0.33333333333333331" alpha="1" colorSpace="custom" customColorSpace="genericGamma22GrayColorSpace"/>
                             </toolbar>
+                            <label hidden="YES" opaque="NO" userInteractionEnabled="NO" contentMode="left" horizontalHuggingPriority="251" verticalHuggingPriority="251" text="downloading" textAlignment="center" lineBreakMode="tailTruncation" baselineAdjustment="alignBaselines" adjustsFontSizeToFit="NO" translatesAutoresizingMaskIntoConstraints="NO" id="i6I-ag-xhJ">
+                                <rect key="frame" x="232" y="131" width="71" height="10"/>
+                                <fontDescription key="fontDescription" type="system" pointSize="8"/>
+                                <color key="textColor" white="0.33333333329999998" alpha="1" colorSpace="custom" customColorSpace="genericGamma22GrayColorSpace"/>
+                                <nil key="highlightedColor"/>
+                            </label>
+                            <label hidden="YES" opaque="NO" userInteractionEnabled="NO" contentMode="left" horizontalHuggingPriority="251" verticalHuggingPriority="251" text="downloading" textAlignment="center" lineBreakMode="tailTruncation" baselineAdjustment="alignBaselines" adjustsFontSizeToFit="NO" translatesAutoresizingMaskIntoConstraints="NO" id="4nE-of-lwx">
+                                <rect key="frame" x="161" y="131" width="71" height="10"/>
+                                <fontDescription key="fontDescription" type="system" pointSize="8"/>
+                                <color key="textColor" white="0.33333333329999998" alpha="1" colorSpace="custom" customColorSpace="genericGamma22GrayColorSpace"/>
+                                <nil key="highlightedColor"/>
+                            </label>
+                            <label hidden="YES" opaque="NO" userInteractionEnabled="NO" contentMode="left" horizontalHuggingPriority="251" verticalHuggingPriority="251" text="downloading" textAlignment="center" lineBreakMode="tailTruncation" baselineAdjustment="alignBaselines" adjustsFontSizeToFit="NO" translatesAutoresizingMaskIntoConstraints="NO" id="yia-6Y-HLX">
+                                <rect key="frame" x="19" y="131" width="71" height="10"/>
+                                <fontDescription key="fontDescription" type="system" pointSize="8"/>
+                                <color key="textColor" white="0.33333333329999998" alpha="1" colorSpace="custom" customColorSpace="genericGamma22GrayColorSpace"/>
+                                <nil key="highlightedColor"/>
+                            </label>
+                            <label hidden="YES" opaque="NO" userInteractionEnabled="NO" contentMode="left" horizontalHuggingPriority="251" verticalHuggingPriority="251" text="downloading" textAlignment="center" lineBreakMode="tailTruncation" baselineAdjustment="alignBaselines" adjustsFontSizeToFit="NO" translatesAutoresizingMaskIntoConstraints="NO" id="M0L-7J-Cbp">
+                                <rect key="frame" x="303" y="131" width="71" height="10"/>
+                                <fontDescription key="fontDescription" type="system" pointSize="8"/>
+                                <color key="textColor" white="0.33333333329999998" alpha="1" colorSpace="custom" customColorSpace="genericGamma22GrayColorSpace"/>
+                                <nil key="highlightedColor"/>
+                            </label>
+                            <label hidden="YES" opaque="NO" userInteractionEnabled="NO" contentMode="left" horizontalHuggingPriority="251" verticalHuggingPriority="251" text="downloading" textAlignment="center" lineBreakMode="tailTruncation" baselineAdjustment="alignBaselines" adjustsFontSizeToFit="NO" translatesAutoresizingMaskIntoConstraints="NO" id="g5X-Me-QvD">
+                                <rect key="frame" x="90" y="131" width="71" height="10"/>
+                                <fontDescription key="fontDescription" type="system" pointSize="8"/>
+                                <color key="textColor" white="0.33333333329999998" alpha="1" colorSpace="custom" customColorSpace="genericGamma22GrayColorSpace"/>
+                                <nil key="highlightedColor"/>
+                            </label>
+                            <button hidden="YES" opaque="NO" contentMode="scaleToFill" contentHorizontalAlignment="center" contentVerticalAlignment="center" buttonType="system" lineBreakMode="middleTruncation" translatesAutoresizingMaskIntoConstraints="NO" id="o9v-9j-MSR">
+                                <rect key="frame" x="19" y="100" width="71" height="41"/>
+                                <state key="normal" title="Button"/>
+                                <buttonConfiguration key="configuration" style="plain"/>
+                                <connections>
+                                    <action selector="priorizeButtonnTapped:" destination="BYZ-38-t0r" eventType="touchUpInside" id="91G-pC-lT6"/>
+                                </connections>
+                            </button>
+                            <button hidden="YES" opaque="NO" contentMode="scaleToFill" contentHorizontalAlignment="center" contentVerticalAlignment="center" buttonType="system" lineBreakMode="middleTruncation" translatesAutoresizingMaskIntoConstraints="NO" id="v2b-qR-PPJ">
+                                <rect key="frame" x="90" y="100" width="71" height="41"/>
+                                <state key="normal" title="Button"/>
+                                <buttonConfiguration key="configuration" style="plain"/>
+                                <connections>
+                                    <action selector="priorizeButtonsTapped:" destination="BYZ-38-t0r" eventType="touchUpInside" id="hMS-m0-8iC"/>
+                                </connections>
+                            </button>
+                            <button hidden="YES" opaque="NO" contentMode="scaleToFill" contentHorizontalAlignment="center" contentVerticalAlignment="center" buttonType="system" lineBreakMode="middleTruncation" translatesAutoresizingMaskIntoConstraints="NO" id="Cur-tW-u1z">
+                                <rect key="frame" x="161" y="100" width="71" height="41"/>
+                                <state key="normal" title="Button"/>
+                                <buttonConfiguration key="configuration" style="plain"/>
+                                <connections>
+                                    <action selector="priorizeButtonmTapped:" destination="BYZ-38-t0r" eventType="touchUpInside" id="yXw-3q-XF4"/>
+                                </connections>
+                            </button>
+                            <button hidden="YES" opaque="NO" contentMode="scaleToFill" contentHorizontalAlignment="center" contentVerticalAlignment="center" buttonType="system" lineBreakMode="middleTruncation" translatesAutoresizingMaskIntoConstraints="NO" id="AKs-vo-fg1">
+                                <rect key="frame" x="232" y="100" width="71" height="41"/>
+                                <state key="normal" title="Button"/>
+                                <buttonConfiguration key="configuration" style="plain"/>
+                                <connections>
+                                    <action selector="priorizeButtonlTapped:" destination="BYZ-38-t0r" eventType="touchUpInside" id="gF1-tg-4uB"/>
+                                </connections>
+                            </button>
+                            <button hidden="YES" opaque="NO" contentMode="scaleToFill" contentHorizontalAlignment="center" contentVerticalAlignment="center" buttonType="system" lineBreakMode="middleTruncation" translatesAutoresizingMaskIntoConstraints="NO" id="U6E-Vx-e7i">
+                                <rect key="frame" x="303" y="100" width="71" height="41"/>
+                                <state key="normal" title="Button"/>
+                                <buttonConfiguration key="configuration" style="plain"/>
+                                <connections>
+                                    <action selector="priorizeButtonxTapped:" destination="BYZ-38-t0r" eventType="touchUpInside" id="ACz-lr-ovy"/>
+                                </connections>
+                            </button>
                         </subviews>
                         <viewLayoutGuide key="safeArea" id="6Tk-OE-BBY"/>
                         <color key="backgroundColor" white="0.0" alpha="1" colorSpace="custom" customColorSpace="genericGamma22GrayColorSpace"/>
                         <gestureRecognizers/>
                         <constraints>
+                            <constraint firstItem="4nE-of-lwx" firstAttribute="leading" secondItem="g5X-Me-QvD" secondAttribute="trailing" id="0yT-ty-SAr"/>
+                            <constraint firstItem="o9v-9j-MSR" firstAttribute="leading" secondItem="yia-6Y-HLX" secondAttribute="leading" id="11R-tb-F38"/>
+                            <constraint firstItem="AKs-vo-fg1" firstAttribute="leading" secondItem="Cur-tW-u1z" secondAttribute="trailing" id="25Q-Dl-gOC"/>
+                            <constraint firstItem="o9v-9j-MSR" firstAttribute="width" secondItem="yia-6Y-HLX" secondAttribute="width" id="2Hf-c7-uMg"/>
+                            <constraint firstItem="i6I-ag-xhJ" firstAttribute="top" secondItem="4nE-of-lwx" secondAttribute="top" id="2lc-YW-Ke3"/>
+                            <constraint firstItem="g5X-Me-QvD" firstAttribute="top" secondItem="yia-6Y-HLX" secondAttribute="top" id="304-91-aVu"/>
+                            <constraint firstItem="Hif-Vv-5nn" firstAttribute="top" secondItem="6Tk-OE-BBY" secondAttribute="top" id="34h-zt-K9b"/>
                             <constraint firstItem="tZo-QP-q2z" firstAttribute="top" secondItem="Gax-hR-qdY" secondAttribute="bottom" constant="1" id="3qP-Re-vUy"/>
+                            <constraint firstItem="v2b-qR-PPJ" firstAttribute="height" secondItem="o9v-9j-MSR" secondAttribute="height" id="3ue-xO-LWX"/>
+                            <constraint firstItem="v2b-qR-PPJ" firstAttribute="bottom" secondItem="o9v-9j-MSR" secondAttribute="bottom" id="4qt-XN-fY8"/>
+                            <constraint firstItem="4nE-of-lwx" firstAttribute="top" secondItem="g5X-Me-QvD" secondAttribute="top" id="6Cd-fH-MR0"/>
+                            <constraint firstItem="v2b-qR-PPJ" firstAttribute="leading" secondItem="o9v-9j-MSR" secondAttribute="trailing" id="6gv-rb-Oit"/>
+                            <constraint firstItem="AKs-vo-fg1" firstAttribute="width" secondItem="Cur-tW-u1z" secondAttribute="width" id="7PG-d9-sgx"/>
+                            <constraint firstItem="9cm-Nq-b3N" firstAttribute="width" secondItem="6Tk-OE-BBY" secondAttribute="width" multiplier="0.903308" id="8Dq-f9-3M2"/>
+                            <constraint firstItem="U6E-Vx-e7i" firstAttribute="leading" secondItem="AKs-vo-fg1" secondAttribute="trailing" id="8p0-eM-Xk2"/>
+                            <constraint firstItem="U6E-Vx-e7i" firstAttribute="width" secondItem="AKs-vo-fg1" secondAttribute="width" id="AS3-dI-Abz"/>
+                            <constraint firstItem="M0L-7J-Cbp" firstAttribute="top" secondItem="i6I-ag-xhJ" secondAttribute="top" id="ATN-by-IXU"/>
+                            <constraint firstItem="U6E-Vx-e7i" firstAttribute="top" secondItem="AKs-vo-fg1" secondAttribute="top" id="BZF-ZQ-7qV"/>
+                            <constraint firstItem="Cur-tW-u1z" firstAttribute="width" secondItem="v2b-qR-PPJ" secondAttribute="width" id="CJ3-Xm-nyK"/>
+                            <constraint firstItem="yia-6Y-HLX" firstAttribute="top" secondItem="9cm-Nq-b3N" secondAttribute="bottom" id="CO6-y9-nuH"/>
+                            <constraint firstItem="i6I-ag-xhJ" firstAttribute="leading" secondItem="4nE-of-lwx" secondAttribute="trailing" id="Ee6-op-pLt"/>
                             <constraint firstItem="Yhe-73-Ryr" firstAttribute="leading" secondItem="6Tk-OE-BBY" secondAttribute="leading" constant="6" id="GB3-HN-lcm"/>
+                            <constraint firstItem="Cur-tW-u1z" firstAttribute="bottom" secondItem="4nE-of-lwx" secondAttribute="bottom" id="HkN-4T-Gcg"/>
+                            <constraint firstItem="Hif-Vv-5nn" firstAttribute="centerX" secondItem="6Tk-OE-BBY" secondAttribute="centerX" id="R2G-VD-gun"/>
+                            <constraint firstItem="i6I-ag-xhJ" firstAttribute="width" secondItem="4nE-of-lwx" secondAttribute="width" id="Ri2-iz-akD"/>
+                            <constraint firstItem="U6E-Vx-e7i" firstAttribute="bottom" secondItem="M0L-7J-Cbp" secondAttribute="bottom" id="RkR-y3-Qb0"/>
                             <constraint firstItem="lFk-9g-4zt" firstAttribute="top" secondItem="Ovr-pp-CtB" secondAttribute="bottom" id="Sqq-Bb-Nvv"/>
                             <constraint firstItem="Ovr-pp-CtB" firstAttribute="leading" secondItem="6Tk-OE-BBY" secondAttribute="leading" constant="5" id="W2U-Xy-2TZ"/>
                             <constraint firstItem="Gax-hR-qdY" firstAttribute="top" secondItem="Yhe-73-Ryr" secondAttribute="bottom" constant="4" id="WvW-3b-L6y"/>
+                            <constraint firstItem="yia-6Y-HLX" firstAttribute="width" secondItem="9cm-Nq-b3N" secondAttribute="width" multiplier="0.2" id="Z57-ES-aRy"/>
+                            <constraint firstItem="4nE-of-lwx" firstAttribute="centerX" secondItem="6Tk-OE-BBY" secondAttribute="centerX" id="aLO-ha-rTr"/>
                             <constraint firstItem="6JI-Fw-yih" firstAttribute="top" secondItem="tZo-QP-q2z" secondAttribute="bottom" constant="4" id="acL-Dc-cLn"/>
                             <constraint firstAttribute="trailing" secondItem="3Rg-UC-ogI" secondAttribute="trailing" id="alq-cZ-iIR"/>
+                            <constraint firstItem="v2b-qR-PPJ" firstAttribute="width" secondItem="o9v-9j-MSR" secondAttribute="width" id="eC2-mv-Azv"/>
+                            <constraint firstItem="4nE-of-lwx" firstAttribute="width" secondItem="g5X-Me-QvD" secondAttribute="width" id="eY7-bW-gV3"/>
                             <constraint firstItem="lFk-9g-4zt" firstAttribute="top" secondItem="Ytf-30-5R8" secondAttribute="bottom" constant="28" id="enE-h4-OrG"/>
+                            <constraint firstItem="9cm-Nq-b3N" firstAttribute="centerX" secondItem="8bC-Xf-vdC" secondAttribute="centerX" id="fOu-qG-Eih"/>
                             <constraint firstItem="Ytf-30-5R8" firstAttribute="leading" secondItem="6Tk-OE-BBY" secondAttribute="leading" constant="8" id="fio-88-KON"/>
                             <constraint firstItem="3Rg-UC-ogI" firstAttribute="bottom" secondItem="6Tk-OE-BBY" secondAttribute="bottom" id="gd1-rI-r2l"/>
+                            <constraint firstItem="M0L-7J-Cbp" firstAttribute="width" secondItem="i6I-ag-xhJ" secondAttribute="width" id="ghp-Zb-ptg"/>
+                            <constraint firstItem="o9v-9j-MSR" firstAttribute="bottom" secondItem="yia-6Y-HLX" secondAttribute="bottom" id="hHT-26-xY9"/>
+                            <constraint firstItem="Cur-tW-u1z" firstAttribute="leading" secondItem="v2b-qR-PPJ" secondAttribute="trailing" id="hhe-mk-hfY"/>
                             <constraint firstItem="tZo-QP-q2z" firstAttribute="leading" secondItem="6Tk-OE-BBY" secondAttribute="leading" constant="6" id="lkb-bl-PkB"/>
+                            <constraint firstItem="AKs-vo-fg1" firstAttribute="bottom" secondItem="i6I-ag-xhJ" secondAttribute="bottom" id="m3A-gh-1Bv"/>
                             <constraint firstItem="6JI-Fw-yih" firstAttribute="leading" secondItem="6Tk-OE-BBY" secondAttribute="leading" constant="5" id="mI7-Wt-LH3"/>
+                            <constraint firstItem="AKs-vo-fg1" firstAttribute="top" secondItem="Cur-tW-u1z" secondAttribute="top" id="mMH-XC-oNm"/>
                             <constraint firstItem="Gax-hR-qdY" firstAttribute="leading" secondItem="6Tk-OE-BBY" secondAttribute="leading" constant="5" id="nUQ-tC-N7o"/>
+                            <constraint firstItem="Cur-tW-u1z" firstAttribute="top" secondItem="v2b-qR-PPJ" secondAttribute="top" id="nuM-zD-v7F"/>
                             <constraint firstItem="3Rg-UC-ogI" firstAttribute="leading" secondItem="6Tk-OE-BBY" secondAttribute="leading" id="os3-Uk-Kum"/>
+                            <constraint firstItem="g5X-Me-QvD" firstAttribute="width" secondItem="yia-6Y-HLX" secondAttribute="width" id="qGJ-bG-ZYi"/>
+                            <constraint firstItem="v2b-qR-PPJ" firstAttribute="top" secondItem="o9v-9j-MSR" secondAttribute="top" id="sKZ-nH-baJ"/>
                             <constraint firstItem="Yhe-73-Ryr" firstAttribute="top" secondItem="lFk-9g-4zt" secondAttribute="bottom" constant="-4" id="sch-Go-Cir"/>
+                            <constraint firstItem="9cm-Nq-b3N" firstAttribute="top" secondItem="Hif-Vv-5nn" secondAttribute="bottom" constant="7" id="spZ-Vn-KPB"/>
+                            <constraint firstItem="o9v-9j-MSR" firstAttribute="top" secondItem="9cm-Nq-b3N" secondAttribute="top" id="wXI-D8-psu"/>
+                            <constraint firstItem="g5X-Me-QvD" firstAttribute="leading" secondItem="yia-6Y-HLX" secondAttribute="trailing" id="xsX-kD-zsC"/>
+                            <constraint firstItem="M0L-7J-Cbp" firstAttribute="leading" secondItem="i6I-ag-xhJ" secondAttribute="trailing" id="z8W-Vc-fhr"/>
                         </constraints>
                         <connections>
                             <outletCollection property="gestureRecognizers" destination="lf1-rc-A2P" appends="YES" id="is6-5B-Ndt"/>
@@ -210,6 +319,11 @@
                     <connections>
                         <outlet property="View0" destination="8bC-Xf-vdC" id="Nwc-r7-Csg"/>
                         <outlet property="activityIndicator" destination="lOR-xu-27i" id="RHs-yl-MIU"/>
+                        <outlet property="downloadingLabell" destination="i6I-ag-xhJ" id="gNL-KH-khT"/>
+                        <outlet property="downloadingLabelm" destination="4nE-of-lwx" id="zlT-1U-l40"/>
+                        <outlet property="downloadingLabeln" destination="yia-6Y-HLX" id="lWP-ow-zeg"/>
+                        <outlet property="downloadingLabels" destination="g5X-Me-QvD" id="IAa-hr-M2v"/>
+                        <outlet property="downloadingLabelx" destination="M0L-7J-Cbp" id="iI5-bm-pg5"/>
                         <outlet property="labelFPS" destination="BZf-Ft-ERG" id="beX-1M-6KR"/>
                         <outlet property="labelName" destination="Hif-Vv-5nn" id="BQr-fg-dFd"/>
                         <outlet property="labelSlider" destination="Ytf-30-5R8" id="wW5-66-erL"/>
@@ -219,6 +333,11 @@
                         <outlet property="labelZoom" destination="vlL-Ye-WgQ" id="OIv-03-1de"/>
                         <outlet property="pauseButtonOutlet" destination="Pes-is-qDX" id="uqc-gB-onZ"/>
                         <outlet property="playButtonOutlet" destination="gHs-cs-nVf" id="YgY-EV-fyY"/>
+                        <outlet property="priorizeButtonl" destination="AKs-vo-fg1" id="I2d-iY-ccr"/>
+                        <outlet property="priorizeButtonm" destination="Cur-tW-u1z" id="Ffr-xb-mpA"/>
+                        <outlet property="priorizeButtonn" destination="o9v-9j-MSR" id="2kl-QX-nrQ"/>
+                        <outlet property="priorizeButtons" destination="v2b-qR-PPJ" id="yr1-yj-vVH"/>
+                        <outlet property="priorizeButtonx" destination="U6E-Vx-e7i" id="MC3-er-GWx"/>
                         <outlet property="segmentedControl" destination="9cm-Nq-b3N" id="RqH-Ft-fGI"/>
                         <outlet property="slider" destination="Ovr-pp-CtB" id="iHy-z4-EIQ"/>
                         <outlet property="sliderConf" destination="Gax-hR-qdY" id="KNe-MM-FBA"/>
diff --git a/YOLO/Utilities/BoundingBoxView.swift b/YOLO/Utilities/BoundingBoxView.swift
index b506545..add81aa 100644
--- a/YOLO/Utilities/BoundingBoxView.swift
+++ b/YOLO/Utilities/BoundingBoxView.swift
@@ -14,66 +14,67 @@ import UIKit
 
 /// Manages the visualization of bounding boxes and associated labels for object detection results.
 class BoundingBoxView {
-    /// The layer that draws the bounding box around a detected object.
-    let shapeLayer: CAShapeLayer
+  /// The layer that draws the bounding box around a detected object.
+  let shapeLayer: CAShapeLayer
 
-    /// The layer that displays the label and confidence score for the detected object.
-    let textLayer: CATextLayer
+  /// The layer that displays the label and confidence score for the detected object.
+  let textLayer: CATextLayer
 
-    /// Initializes a new BoundingBoxView with configured shape and text layers.
-    init() {
-        shapeLayer = CAShapeLayer()
-        shapeLayer.fillColor = UIColor.clear.cgColor  // No fill to only show the bounding outline
-        shapeLayer.lineWidth = 4  // Set the stroke line width
-        shapeLayer.isHidden = true  // Initially hidden; shown when a detection occurs
+  /// Initializes a new BoundingBoxView with configured shape and text layers.
+  init() {
+    shapeLayer = CAShapeLayer()
+    shapeLayer.fillColor = UIColor.clear.cgColor  // No fill to only show the bounding outline
+    shapeLayer.lineWidth = 4  // Set the stroke line width
+    shapeLayer.isHidden = true  // Initially hidden; shown when a detection occurs
 
-        textLayer = CATextLayer()
-        textLayer.isHidden = true  // Initially hidden; shown with label when a detection occurs
-        textLayer.contentsScale = UIScreen.main.scale  // Ensure the text is sharp on retina displays
-        textLayer.fontSize = 14  // Set font size for the label text
-        textLayer.font = UIFont(name: "Avenir", size: textLayer.fontSize)  // Use Avenir font for labels
-        textLayer.alignmentMode = .center  // Center-align the text within the layer
-    }
+    textLayer = CATextLayer()
+    textLayer.isHidden = true  // Initially hidden; shown with label when a detection occurs
+    textLayer.contentsScale = UIScreen.main.scale  // Ensure the text is sharp on retina displays
+    textLayer.fontSize = 14  // Set font size for the label text
+    textLayer.font = UIFont(name: "Avenir", size: textLayer.fontSize)  // Use Avenir font for labels
+    textLayer.alignmentMode = .center  // Center-align the text within the layer
+  }
 
-    /// Adds the bounding box and text layers to a specified parent layer.
-    /// - Parameter parent: The CALayer to which the bounding box and text layers will be added.
-    func addToLayer(_ parent: CALayer) {
-        parent.addSublayer(shapeLayer)
-        parent.addSublayer(textLayer)
-    }
+  /// Adds the bounding box and text layers to a specified parent layer.
+  /// - Parameter parent: The CALayer to which the bounding box and text layers will be added.
+  func addToLayer(_ parent: CALayer) {
+    parent.addSublayer(shapeLayer)
+    parent.addSublayer(textLayer)
+  }
 
-    /// Updates the bounding box and label to be visible with specified properties.
-    /// - Parameters:
-    ///   - frame: The CGRect frame defining the bounding box's size and position.
-    ///   - label: The text label to display (e.g., object class and confidence).
-    ///   - color: The color of the bounding box stroke and label background.
-    ///   - alpha: The opacity level for the bounding box stroke and label background.
-    func show(frame: CGRect, label: String, color: UIColor, alpha: CGFloat) {
-        CATransaction.setDisableActions(true)  // Disable implicit animations
+  /// Updates the bounding box and label to be visible with specified properties.
+  /// - Parameters:
+  ///   - frame: The CGRect frame defining the bounding box's size and position.
+  ///   - label: The text label to display (e.g., object class and confidence).
+  ///   - color: The color of the bounding box stroke and label background.
+  ///   - alpha: The opacity level for the bounding box stroke and label background.
+  func show(frame: CGRect, label: String, color: UIColor, alpha: CGFloat) {
+    CATransaction.setDisableActions(true)  // Disable implicit animations
 
-        let path = UIBezierPath(roundedRect: frame, cornerRadius: 6.0)  // Rounded rectangle for the bounding box
-        shapeLayer.path = path.cgPath
-        shapeLayer.strokeColor = color.withAlphaComponent(alpha).cgColor  // Apply color and alpha to the stroke
-        shapeLayer.isHidden = false  // Make the shape layer visible
+    let path = UIBezierPath(roundedRect: frame, cornerRadius: 6.0)  // Rounded rectangle for the bounding box
+    shapeLayer.path = path.cgPath
+    shapeLayer.strokeColor = color.withAlphaComponent(alpha).cgColor  // Apply color and alpha to the stroke
+    shapeLayer.isHidden = false  // Make the shape layer visible
 
-        textLayer.string = label  // Set the label text
-        textLayer.backgroundColor = color.withAlphaComponent(alpha).cgColor  // Apply color and alpha to the background
-        textLayer.isHidden = false  // Make the text layer visible
-        textLayer.foregroundColor = UIColor.white.withAlphaComponent(alpha).cgColor  // Set text color
+    textLayer.string = label  // Set the label text
+    textLayer.backgroundColor = color.withAlphaComponent(alpha).cgColor  // Apply color and alpha to the background
+    textLayer.isHidden = false  // Make the text layer visible
+    textLayer.foregroundColor = UIColor.white.withAlphaComponent(alpha).cgColor  // Set text color
 
-        // Calculate the text size and position based on the label content
-        let attributes = [NSAttributedString.Key.font: textLayer.font as Any]
-        let textRect = label.boundingRect(with: CGSize(width: 400, height: 100),
-                options: .truncatesLastVisibleLine,
-                attributes: attributes, context: nil)
-        let textSize = CGSize(width: textRect.width + 12, height: textRect.height)  // Add padding to the text size
-        let textOrigin = CGPoint(x: frame.origin.x - 2, y: frame.origin.y - textSize.height - 2)  // Position above the bounding box
-        textLayer.frame = CGRect(origin: textOrigin, size: textSize)  // Set the text layer frame
-    }
+    // Calculate the text size and position based on the label content
+    let attributes = [NSAttributedString.Key.font: textLayer.font as Any]
+    let textRect = label.boundingRect(
+      with: CGSize(width: 400, height: 100),
+      options: .truncatesLastVisibleLine,
+      attributes: attributes, context: nil)
+    let textSize = CGSize(width: textRect.width + 12, height: textRect.height)  // Add padding to the text size
+    let textOrigin = CGPoint(x: frame.origin.x - 2, y: frame.origin.y - textSize.height - 2)  // Position above the bounding box
+    textLayer.frame = CGRect(origin: textOrigin, size: textSize)  // Set the text layer frame
+  }
 
-    /// Hides the bounding box and text layers.
-    func hide() {
-        shapeLayer.isHidden = true
-        textLayer.isHidden = true
-    }
+  /// Hides the bounding box and text layers.
+  func hide() {
+    shapeLayer.isHidden = true
+    textLayer.isHidden = true
+  }
 }
diff --git a/YOLO/Utilities/ModelChacheManager.swift b/YOLO/Utilities/ModelChacheManager.swift
new file mode 100644
index 0000000..9ef3d6a
--- /dev/null
+++ b/YOLO/Utilities/ModelChacheManager.swift
@@ -0,0 +1,327 @@
+import CoreML
+import Foundation
+import ZIPFoundation
+
+class ModelCacheManager {
+  static let shared = ModelCacheManager()
+  var modelCache: [String: MLModel] = [:]
+  private var accessOrder: [String] = []
+  private let cacheLimit: Int = 3
+  private var currentSelectedModelKey: String?
+
+  private init() {
+    //        loadBundledModel()
+  }
+
+  private func getRemoteURL(for key: String) -> URL? {
+    for (fileName, url) in fileMappings {
+      if fileName == key {
+        return url
+      }
+    }
+    return nil
+  }
+
+  func loadBundledModel() {
+    if let url = getModelFileURL(fileName: "yolov8m"),
+      let bundledModel = try? MLModel(contentsOf: url)
+    {
+      addModelToCache(bundledModel, for: "yolov8m")
+      let documentsURL = getDocumentsDirectory()
+      let destinationURL = documentsURL.appendingPathComponent("yolov8m").appendingPathExtension(
+        "mlmodelc")
+
+      do {
+        if !FileManager.default.fileExists(atPath: destinationURL.path) {
+          try FileManager.default.copyItem(at: url, to: destinationURL)
+          print("File copied to documents directory: \(destinationURL.path)")
+        } else {
+          print("File already exists in documents directory: \(destinationURL.path)")
+        }
+      } catch {
+        print("Error copying file: \(error)")
+      }
+    } else {
+      print("Failed to load bundled model")
+    }
+  }
+
+  func loadLocalModel(key: String, completion: @escaping (MLModel?, String) -> Void) {
+    if let cachedModel = modelCache[key] {
+      if let index = accessOrder.firstIndex(of: key) {
+        accessOrder.remove(at: index)
+      }
+      accessOrder.append(key)
+      completion(cachedModel, key)
+      return
+    }
+
+    let modelURL = getDocumentsDirectory().appendingPathComponent(key).appendingPathExtension(
+      "mlmodelc")
+    if FileManager.default.fileExists(atPath: modelURL.path) {
+      do {
+        let model = try MLModel(contentsOf: modelURL)
+        addModelToCache(model, for: key)
+        completion(model, key)
+      } catch let error {
+        print(error)
+      }
+    }
+  }
+
+  func loadModel(
+    from fileName: String, remoteURL: URL, key: String,
+    completion: @escaping (MLModel?, String) -> Void
+  ) {
+    if let cachedModel = modelCache[key] {
+      if let index = accessOrder.firstIndex(of: key) {
+        accessOrder.remove(at: index)
+      }
+      accessOrder.append(key)
+      completion(cachedModel, key)
+      return
+    }
+
+    let modelURL = getDocumentsDirectory().appendingPathComponent(key).appendingPathExtension(
+      "mlmodelc")
+    if FileManager.default.fileExists(atPath: modelURL.path) {
+      loadLocalModel(key: key) { model, key in
+        completion(model, key)
+      }
+    } else {
+      ModelDownloadManager.shared.startDownload(
+        url: remoteURL, fileName: fileName, key: key, completion: completion)
+    }
+  }
+
+  private func compileAndLoadDownloadedModel(
+    from url: URL, key: String, completion: @escaping (MLModel?) -> Void
+  ) {
+    DispatchQueue.global(qos: .userInitiated).async {
+      do {
+
+        let compiledModelURL = try MLModel.compileModel(at: url)
+        let destinationURL = self.getDocumentsDirectory().appendingPathComponent(
+          compiledModelURL.lastPathComponent)
+
+        // Move compiled model to desired location
+        try FileManager.default.moveItem(at: compiledModelURL, to: destinationURL)
+        let model = try MLModel(contentsOf: destinationURL)
+        DispatchQueue.main.async {
+          self.addModelToCache(model, for: key)
+          completion(model)
+        }
+      } catch {
+        print("Failed to load model: \(error)")
+        DispatchQueue.main.async {
+          completion(nil)
+        }
+      }
+    }
+  }
+
+  func addModelToCache(_ model: MLModel, for key: String) {
+    if modelCache.count >= cacheLimit {
+      let oldKey = accessOrder.removeFirst()
+      modelCache.removeValue(forKey: oldKey)
+    }
+    modelCache[key] = model
+    accessOrder.append(key)
+  }
+
+  func isModelDownloaded(key: String) -> Bool {
+    let modelURL = getDocumentsDirectory().appendingPathComponent(key).appendingPathExtension(
+      "mlmodelc")
+    return FileManager.default.fileExists(atPath: modelURL.path)
+  }
+
+  private func getDocumentsDirectory() -> URL {
+    return FileManager.default.urls(for: .documentDirectory, in: .userDomainMask)[0]
+  }
+
+  func prioritizeDownload(for fileName: String, completion: @escaping (MLModel?, String) -> Void) {
+    ModelDownloadManager.shared.prioritizeDownload(for: fileName, completion: completion)
+  }
+
+  func setCurrentSelectedModelKey(_ key: String) {
+    currentSelectedModelKey = key
+  }
+
+  func getCurrentSelectedModelKey() -> String? {
+    return currentSelectedModelKey
+  }
+}
+
+class ModelDownloadManager: NSObject {
+  static let shared = ModelDownloadManager()
+  private var downloadTasks: [URLSessionDownloadTask: (url: URL, key: String)] = [:]
+  private var priorityTask: URLSessionDownloadTask?
+
+  private override init() {}
+
+  func startDownload(
+    url: URL, fileName: String, key: String, completion: @escaping (MLModel?, String) -> Void
+  ) {
+    let session = URLSession(configuration: .default, delegate: self, delegateQueue: nil)
+    let downloadTask = session.downloadTask(with: url)
+    let destinationURL = getDocumentsDirectory().appendingPathComponent(fileName)
+    downloadTasks[downloadTask] = (url: destinationURL, key: key)
+    downloadTask.resume()
+    downloadCompletionHandlers[downloadTask] = completion
+  }
+
+  private var downloadCompletionHandlers: [URLSessionDownloadTask: (MLModel?, String) -> Void] = [:]
+
+  private func getDocumentsDirectory() -> URL {
+    return FileManager.default.urls(for: .documentDirectory, in: .userDomainMask)[0]
+  }
+
+  func prioritizeDownload(for fileName: String, completion: @escaping (MLModel?, String) -> Void) {
+    for (task, urlKeyPair) in downloadTasks {
+      if urlKeyPair.url.lastPathComponent.contains(fileName) {
+        task.cancel(byProducingResumeData: { resumeData in
+          if let resumeData = resumeData {
+            let session = URLSession(configuration: .default, delegate: self, delegateQueue: nil)
+            let priorityDownloadTask = session.downloadTask(withResumeData: resumeData)
+            priorityDownloadTask.priority = URLSessionTask.highPriority
+            self.downloadTasks[priorityDownloadTask] = urlKeyPair
+            self.downloadCompletionHandlers[priorityDownloadTask] = completion
+            self.priorityTask = priorityDownloadTask
+            priorityDownloadTask.resume()
+          } else {
+            let session = URLSession(configuration: .default, delegate: self, delegateQueue: nil)
+            let priorityDownloadTask = session.downloadTask(with: task.originalRequest!)
+            priorityDownloadTask.priority = URLSessionTask.highPriority
+            self.downloadTasks[priorityDownloadTask] = urlKeyPair
+            self.downloadCompletionHandlers[priorityDownloadTask] = completion
+            self.priorityTask = priorityDownloadTask
+            priorityDownloadTask.resume()
+          }
+        })
+        break
+      }
+    }
+  }
+
+  func cancelCurrentDownload() {
+    priorityTask?.cancel()
+    priorityTask = nil
+  }
+}
+
+extension ModelDownloadManager: URLSessionDownloadDelegate {
+  func urlSession(
+    _ session: URLSession, downloadTask: URLSessionDownloadTask,
+    didFinishDownloadingTo location: URL
+  ) {
+    guard let destinationURL = downloadTasks[downloadTask]?.url,
+      let key = downloadTasks[downloadTask]?.key
+    else { return }
+    do {
+      let zipURL = destinationURL
+      print("zipURL: \(zipURL)")
+      if fileExists(at: zipURL) {
+        try FileManager.default.removeItem(at: zipURL)
+      }
+      try FileManager.default.moveItem(at: location, to: zipURL)
+      downloadTasks.removeValue(forKey: downloadTask)
+      let unzipDestinationURL = destinationURL.deletingPathExtension()
+
+      do {
+        try FileManager.default.unzipItem(at: zipURL, to: getDocumentsDirectory(), skipCRC32: true)
+        let modelURL = unzipDestinationURL
+        print("modelURL: \(modelURL)")
+        loadModel(from: modelURL, key: key) { model in
+          self.downloadCompletionHandlers[downloadTask]?(model, key)
+          self.downloadCompletionHandlers.removeValue(forKey: downloadTask)
+        }
+      } catch {
+        print("Extraction of ZIP archive failed with error: \(error)")
+        self.downloadCompletionHandlers[downloadTask]?(nil, key)
+        self.downloadCompletionHandlers.removeValue(forKey: downloadTask)
+      }
+    } catch {
+      print("Error moving downloaded file: \(error)")
+      self.downloadCompletionHandlers[downloadTask]?(nil, key)
+      self.downloadCompletionHandlers.removeValue(forKey: downloadTask)
+    }
+  }
+
+  private func loadModel(from url: URL, key: String, completion: @escaping (MLModel?) -> Void) {
+    DispatchQueue.global(qos: .userInitiated).async {
+      do {
+        let compiledModelURL = try MLModel.compileModel(at: url)
+        let model = try MLModel(contentsOf: compiledModelURL)
+        let localModelURL = self.getDocumentsDirectory().appendingPathComponent(key)
+          .appendingPathExtension("mlmodelc")
+        ModelCacheManager.shared.addModelToCache(model, for: key)
+        try FileManager.default.moveItem(at: compiledModelURL, to: localModelURL)
+        print("model copied to document directory")
+        DispatchQueue.main.async {
+          completion(model)
+        }
+      } catch {
+        print("Failed to load model: \(error)")
+        DispatchQueue.main.async {
+          completion(nil)
+        }
+      }
+    }
+  }
+}
+
+class ModelFileManager {
+  static let shared = ModelFileManager()
+
+  private init() {}
+
+  func deleteAllDownloadedModels() {
+    let fileManager = FileManager.default
+    let documentsDirectory = getDocumentsDirectory()
+    do {
+      let fileURLs = try fileManager.contentsOfDirectory(
+        at: documentsDirectory, includingPropertiesForKeys: nil)
+      for fileURL in fileURLs {
+        if fileURL.pathExtension == "mlmodel" || fileURL.pathExtension == "mlmodelc"
+          || fileURL.pathExtension == "mlpackage"
+        {
+          try fileManager.removeItem(at: fileURL)
+          print("Deleted file: \(fileURL.lastPathComponent)")
+        }
+      }
+    } catch {
+      print("Error deleting files: \(error)")
+    }
+  }
+
+  private func getDocumentsDirectory() -> URL {
+    return FileManager.default.urls(for: .documentDirectory, in: .userDomainMask)[0]
+  }
+}
+
+func getModelFileURL(fileName: String) -> URL? {
+  let bundle = Bundle.main
+  if let fileURL = bundle.url(forResource: fileName, withExtension: "mlmodelc") {
+    return fileURL
+  }
+  return nil
+}
+
+func fileExists(at url: URL) -> Bool {
+  let fileManager = FileManager.default
+  return fileManager.fileExists(atPath: url.path)
+}
+
+extension URL {
+  func changingFileExtension(to newExtension: String) -> URL? {
+    var urlString = self.absoluteString
+
+    if let range = urlString.range(of: "\\.[^./]*$", options: .regularExpression) {
+      urlString.replaceSubrange(range, with: ".\(newExtension)")
+    } else {
+      urlString.append(".\(newExtension)")
+    }
+
+    return URL(string: urlString)
+  }
+}
diff --git a/YOLO/Utilities/Models.swift b/YOLO/Utilities/Models.swift
new file mode 100644
index 0000000..3b8d804
--- /dev/null
+++ b/YOLO/Utilities/Models.swift
@@ -0,0 +1,44 @@
+import Foundation
+
+let presetModels = ["yolov8n", "yolov8s", "yolov8m", "yolov8l", "yolov8x"]
+
+let firstModels = [
+  "detect": (
+    "yolov8m",
+    URL(
+      string:
+        "https://firebasestorage.googleapis.com/v0/b/sincere-nirvana-292404.appspot.com/o/yolov8m.mlpackage.zip?alt=media&token=1ba085c3-dc38-48ec-9806-1d5cbdc291cc"
+    )!
+  )
+]
+
+let fileMappings = [
+  (
+    "yolov8n",
+    URL(
+      string:
+        "https://firebasestorage.googleapis.com/v0/b/sincere-nirvana-292404.appspot.com/o/yolov8n.mlpackage.zip?alt=media&token=80a62f8e-96f1-4355-8857-51b4ecd6f2e3"
+    )!
+  ),
+  (
+    "yolov8s",
+    URL(
+      string:
+        "https://firebasestorage.googleapis.com/v0/b/sincere-nirvana-292404.appspot.com/o/yolov8s.mlpackage.zip?alt=media&token=8f18404a-6c33-49f6-b89a-a3b69a4aa0bb"
+    )!
+  ),
+  (
+    "yolov8l",
+    URL(
+      string:
+        "https://firebasestorage.googleapis.com/v0/b/sincere-nirvana-292404.appspot.com/o/yolov8l.mlpackage.zip?alt=media&token=393c644e-b848-4c97-a4d0-61cfa6ffc307"
+    )!
+  ),
+  (
+    "yolov8x",
+    URL(
+      string:
+        "https://firebasestorage.googleapis.com/v0/b/sincere-nirvana-292404.appspot.com/o/yolov8x.mlpackage.zip?alt=media&token=f3074450-00bd-4b44-962d-f4e122d73e6b"
+    )!
+  ),
+]
diff --git a/YOLO/Utilities/ThresholdProvider.swift b/YOLO/Utilities/ThresholdProvider.swift
index 53702d1..22c8d6a 100644
--- a/YOLO/Utilities/ThresholdProvider.swift
+++ b/YOLO/Utilities/ThresholdProvider.swift
@@ -14,29 +14,29 @@ import CoreML
 
 /// Provides custom IoU and confidence thresholds for adjusting model predictions.
 class ThresholdProvider: MLFeatureProvider {
-    /// Stores IoU and confidence thresholds as MLFeatureValue objects.
-    var values: [String: MLFeatureValue]
+  /// Stores IoU and confidence thresholds as MLFeatureValue objects.
+  var values: [String: MLFeatureValue]
 
-    /// The set of feature names provided by this provider.
-    var featureNames: Set<String> {
-        return Set(values.keys)
-    }
+  /// The set of feature names provided by this provider.
+  var featureNames: Set<String> {
+    return Set(values.keys)
+  }
 
-    /// Initializes the provider with specified IoU and confidence thresholds.
-    /// - Parameters:
-    ///   - iouThreshold: The IoU threshold for determining object overlap.
-    ///   - confidenceThreshold: The minimum confidence for considering a detection valid.
-    init(iouThreshold: Double = 0.45, confidenceThreshold: Double = 0.25) {
-        values = [
-            "iouThreshold": MLFeatureValue(double: iouThreshold),
-            "confidenceThreshold": MLFeatureValue(double: confidenceThreshold)
-        ]
-    }
+  /// Initializes the provider with specified IoU and confidence thresholds.
+  /// - Parameters:
+  ///   - iouThreshold: The IoU threshold for determining object overlap.
+  ///   - confidenceThreshold: The minimum confidence for considering a detection valid.
+  init(iouThreshold: Double = 0.45, confidenceThreshold: Double = 0.25) {
+    values = [
+      "iouThreshold": MLFeatureValue(double: iouThreshold),
+      "confidenceThreshold": MLFeatureValue(double: confidenceThreshold),
+    ]
+  }
 
-    /// Returns the feature value for the given feature name.
-    /// - Parameter featureName: The name of the feature.
-    /// - Returns: The MLFeatureValue object corresponding to the feature name.
-    func featureValue(for featureName: String) -> MLFeatureValue? {
-        return values[featureName]
-    }
+  /// Returns the feature value for the given feature name.
+  /// - Parameter featureName: The name of the feature.
+  /// - Returns: The MLFeatureValue object corresponding to the feature name.
+  func featureValue(for featureName: String) -> MLFeatureValue? {
+    return values[featureName]
+  }
 }
diff --git a/YOLO/VideoCapture.swift b/YOLO/VideoCapture.swift
index 79aaf99..82e7fe9 100644
--- a/YOLO/VideoCapture.swift
+++ b/YOLO/VideoCapture.swift
@@ -11,123 +11,134 @@
 //  the capture session. It also provides methods to start and stop video capture and delivers captured frames
 //  to a delegate implementing the VideoCaptureDelegate protocol.
 
-
 import AVFoundation
 import CoreVideo
 import UIKit
 
 // Defines the protocol for handling video frame capture events.
 public protocol VideoCaptureDelegate: AnyObject {
-    func videoCapture(_ capture: VideoCapture, didCaptureVideoFrame: CMSampleBuffer)
+  func videoCapture(_ capture: VideoCapture, didCaptureVideoFrame: CMSampleBuffer)
 }
 
 // Identifies the best available camera device based on user preferences and device capabilities.
 func bestCaptureDevice() -> AVCaptureDevice {
-    if UserDefaults.standard.bool(forKey: "use_telephoto"), let device = AVCaptureDevice.default(.builtInTelephotoCamera, for: .video, position: .back) {
-        return device
-    } else if let device = AVCaptureDevice.default(.builtInDualCamera, for: .video, position: .back) {
-        return device
-    } else if let device = AVCaptureDevice.default(.builtInWideAngleCamera, for: .video, position: .back) {
-        return device
-    } else {
-        fatalError("Expected back camera device is not available.")
-    }
+  if UserDefaults.standard.bool(forKey: "use_telephoto"),
+    let device = AVCaptureDevice.default(.builtInTelephotoCamera, for: .video, position: .back)
+  {
+    return device
+  } else if let device = AVCaptureDevice.default(.builtInDualCamera, for: .video, position: .back) {
+    return device
+  } else if let device = AVCaptureDevice.default(
+    .builtInWideAngleCamera, for: .video, position: .back)
+  {
+    return device
+  } else {
+    fatalError("Expected back camera device is not available.")
+  }
 }
 
 public class VideoCapture: NSObject {
-    public var previewLayer: AVCaptureVideoPreviewLayer?
-    public weak var delegate: VideoCaptureDelegate?
-
-    let captureDevice = bestCaptureDevice()
-    let captureSession = AVCaptureSession()
-    let videoOutput = AVCaptureVideoDataOutput()
-    var cameraOutput = AVCapturePhotoOutput()
-    let queue = DispatchQueue(label: "camera-queue")
-
-    // Configures the camera and capture session with optional session presets.
-    public func setUp(sessionPreset: AVCaptureSession.Preset = .hd1280x720, completion: @escaping (Bool) -> Void) {
-        queue.async {
-            let success = self.setUpCamera(sessionPreset: sessionPreset)
-            DispatchQueue.main.async {
-                completion(success)
-            }
-        }
+  public var previewLayer: AVCaptureVideoPreviewLayer?
+  public weak var delegate: VideoCaptureDelegate?
+
+  let captureDevice = bestCaptureDevice()
+  let captureSession = AVCaptureSession()
+  let videoOutput = AVCaptureVideoDataOutput()
+  var cameraOutput = AVCapturePhotoOutput()
+  let queue = DispatchQueue(label: "camera-queue")
+
+  // Configures the camera and capture session with optional session presets.
+  public func setUp(
+    sessionPreset: AVCaptureSession.Preset = .hd1280x720, completion: @escaping (Bool) -> Void
+  ) {
+    queue.async {
+      let success = self.setUpCamera(sessionPreset: sessionPreset)
+      DispatchQueue.main.async {
+        completion(success)
+      }
     }
+  }
+
+  // Internal method to configure camera inputs, outputs, and session properties.
+  private func setUpCamera(sessionPreset: AVCaptureSession.Preset) -> Bool {
+    captureSession.beginConfiguration()
+    captureSession.sessionPreset = sessionPreset
 
-    // Internal method to configure camera inputs, outputs, and session properties.
-    private func setUpCamera(sessionPreset: AVCaptureSession.Preset) -> Bool {
-        captureSession.beginConfiguration()
-        captureSession.sessionPreset = sessionPreset
-
-        guard let videoInput = try? AVCaptureDeviceInput(device: captureDevice) else {
-            return false
-        }
-
-        if captureSession.canAddInput(videoInput) {
-            captureSession.addInput(videoInput)
-        }
-
-        let previewLayer = AVCaptureVideoPreviewLayer(session: captureSession)
-        previewLayer.videoGravity = .resizeAspectFill
-        previewLayer.connection?.videoOrientation = .portrait
-        self.previewLayer = previewLayer
-
-        let settings: [String: Any] = [
-            kCVPixelBufferPixelFormatTypeKey as String: NSNumber(value: kCVPixelFormatType_32BGRA)
-        ]
-
-        videoOutput.videoSettings = settings
-        videoOutput.alwaysDiscardsLateVideoFrames = true
-        videoOutput.setSampleBufferDelegate(self, queue: queue)
-        if captureSession.canAddOutput(videoOutput) {
-            captureSession.addOutput(videoOutput)
-        }
-
-        if captureSession.canAddOutput(cameraOutput) {
-            captureSession.addOutput(cameraOutput)
-        }
-
-        videoOutput.connection(with: .video)?.videoOrientation = .portrait
-
-        do {
-            try captureDevice.lockForConfiguration()
-            captureDevice.focusMode = .continuousAutoFocus
-            captureDevice.focusPointOfInterest = CGPoint(x: 0.5, y: 0.5)
-            captureDevice.exposureMode = .continuousAutoExposure
-            captureDevice.unlockForConfiguration()
-        } catch {
-            print("Unable to configure the capture device.")
-            return false
-        }
-
-        captureSession.commitConfiguration()
-        return true
+    guard let videoInput = try? AVCaptureDeviceInput(device: captureDevice) else {
+      return false
     }
 
-    // Starts the video capture session.
-    public func start() {
-        if !captureSession.isRunning {
-            DispatchQueue.global(qos: .userInitiated).async { [weak self] in
-                self?.captureSession.startRunning()
-            }
-        }
+    if captureSession.canAddInput(videoInput) {
+      captureSession.addInput(videoInput)
     }
 
-    // Stops the video capture session.
-    public func stop() {
-        if captureSession.isRunning {
-            captureSession.stopRunning()
-        }
+    let previewLayer = AVCaptureVideoPreviewLayer(session: captureSession)
+    previewLayer.videoGravity = .resizeAspectFill
+    previewLayer.connection?.videoOrientation = .portrait
+    self.previewLayer = previewLayer
+
+    let settings: [String: Any] = [
+      kCVPixelBufferPixelFormatTypeKey as String: NSNumber(value: kCVPixelFormatType_32BGRA)
+    ]
+
+    videoOutput.videoSettings = settings
+    videoOutput.alwaysDiscardsLateVideoFrames = true
+    videoOutput.setSampleBufferDelegate(self, queue: queue)
+    if captureSession.canAddOutput(videoOutput) {
+      captureSession.addOutput(videoOutput)
     }
-}
 
-// Extension to handle AVCaptureVideoDataOutputSampleBufferDelegate events.
-extension VideoCapture: AVCaptureVideoDataOutputSampleBufferDelegate {
-    public func captureOutput(_ output: AVCaptureOutput, didOutput sampleBuffer: CMSampleBuffer, from connection: AVCaptureConnection) {
-        delegate?.videoCapture(self, didCaptureVideoFrame: sampleBuffer)
+    if captureSession.canAddOutput(cameraOutput) {
+      captureSession.addOutput(cameraOutput)
     }
 
-    public func captureOutput(_ output: AVCaptureOutput, didDrop sampleBuffer: CMSampleBuffer, from connection: AVCaptureConnection) {
-        // Optionally handle dropped frames, e.g., due to full buffer.
+    videoOutput.connection(with: .video)?.videoOrientation = .portrait
+
+    do {
+      try captureDevice.lockForConfiguration()
+      captureDevice.focusMode = .continuousAutoFocus
+      captureDevice.focusPointOfInterest = CGPoint(x: 0.5, y: 0.5)
+      captureDevice.exposureMode = .continuousAutoExposure
+      captureDevice.unlockForConfiguration()
+    } catch {
+      print("Unable to configure the capture device.")
+      return false
     }
+
+    captureSession.commitConfiguration()
+    return true
+  }
+
+  // Starts the video capture session.
+  public func start() {
+    if !captureSession.isRunning {
+      DispatchQueue.global(qos: .userInitiated).async { [weak self] in
+        self?.captureSession.startRunning()
+      }
+    }
+  }
+
+  // Stops the video capture session.
+  public func stop() {
+    if captureSession.isRunning {
+      captureSession.stopRunning()
+    }
+  }
+}
+
+// Extension to handle AVCaptureVideoDataOutputSampleBufferDelegate events.
+extension VideoCapture: AVCaptureVideoDataOutputSampleBufferDelegate {
+  public func captureOutput(
+    _ output: AVCaptureOutput, didOutput sampleBuffer: CMSampleBuffer,
+    from connection: AVCaptureConnection
+  ) {
+    delegate?.videoCapture(self, didCaptureVideoFrame: sampleBuffer)
+  }
+
+  public func captureOutput(
+    _ output: AVCaptureOutput, didDrop sampleBuffer: CMSampleBuffer,
+    from connection: AVCaptureConnection
+  ) {
+    // Optionally handle dropped frames, e.g., due to full buffer.
+  }
 }
diff --git a/YOLO/ViewController.swift b/YOLO/ViewController.swift
index 9ab212e..cbec303 100644
--- a/YOLO/ViewController.swift
+++ b/YOLO/ViewController.swift
@@ -12,578 +12,834 @@
 //  the device's camera.
 
 import AVFoundation
-import CoreMedia
 import CoreML
+import CoreMedia
 import UIKit
 import Vision
 
-var mlModel = try! yolov8m(configuration: .init()).model
+var mlModel: MLModel!
 
 class ViewController: UIViewController {
-    @IBOutlet var videoPreview: UIView!
-    @IBOutlet var View0: UIView!
-    @IBOutlet var segmentedControl: UISegmentedControl!
-    @IBOutlet var playButtonOutlet: UIBarButtonItem!
-    @IBOutlet var pauseButtonOutlet: UIBarButtonItem!
-    @IBOutlet var slider: UISlider!
-    @IBOutlet var sliderConf: UISlider!
-    @IBOutlet var sliderIoU: UISlider!
-    @IBOutlet weak var labelName: UILabel!
-    @IBOutlet weak var labelFPS: UILabel!
-    @IBOutlet weak var labelZoom: UILabel!
-    @IBOutlet weak var labelVersion: UILabel!
-    @IBOutlet weak var labelSlider: UILabel!
-    @IBOutlet weak var labelSliderConf: UILabel!
-    @IBOutlet weak var labelSliderIoU: UILabel!
-    @IBOutlet weak var activityIndicator: UIActivityIndicatorView!
-
-    let selection = UISelectionFeedbackGenerator()
-    var detector = try! VNCoreMLModel(for: mlModel)
-    var session: AVCaptureSession!
-    var videoCapture: VideoCapture!
-    var currentBuffer: CVPixelBuffer?
-    var framesDone = 0
-    var t0 = 0.0  // inference start
-    var t1 = 0.0  // inference dt
-    var t2 = 0.0  // inference dt smoothed
-    var t3 = CACurrentMediaTime()  // FPS start
-    var t4 = 0.0  // FPS dt smoothed
-    // var cameraOutput: AVCapturePhotoOutput!
-
-    // Developer mode
-    let developerMode = UserDefaults.standard.bool(forKey: "developer_mode")   // developer mode selected in settings
-    let save_detections = false  // write every detection to detections.txt
-    let save_frames = false  // write every frame to frames.txt
-
-    lazy var visionRequest: VNCoreMLRequest = {
-        let request = VNCoreMLRequest(model: detector, completionHandler: {
-            [weak self] request, error in
-            self?.processObservations(for: request, error: error)
-        })
-        // NOTE: BoundingBoxView object scaling depends on request.imageCropAndScaleOption https://developer.apple.com/documentation/vision/vnimagecropandscaleoption
-        request.imageCropAndScaleOption = .scaleFill  // .scaleFit, .scaleFill, .centerCrop
-        return request
-    }()
-
-    override func viewDidLoad() {
-        super.viewDidLoad()
-        slider.value = 30
-        setLabels()
+  @IBOutlet var videoPreview: UIView!
+  @IBOutlet var View0: UIView!
+  @IBOutlet var segmentedControl: UISegmentedControl!
+  @IBOutlet var playButtonOutlet: UIBarButtonItem!
+  @IBOutlet var pauseButtonOutlet: UIBarButtonItem!
+  @IBOutlet var slider: UISlider!
+  @IBOutlet var sliderConf: UISlider!
+  @IBOutlet var sliderIoU: UISlider!
+  @IBOutlet weak var labelName: UILabel!
+  @IBOutlet weak var labelFPS: UILabel!
+  @IBOutlet weak var labelZoom: UILabel!
+  @IBOutlet weak var labelVersion: UILabel!
+  @IBOutlet weak var labelSlider: UILabel!
+  @IBOutlet weak var labelSliderConf: UILabel!
+  @IBOutlet weak var labelSliderIoU: UILabel!
+  @IBOutlet weak var activityIndicator: UIActivityIndicatorView!
+  @IBOutlet weak var downloadingLabeln: UILabel!
+  @IBOutlet weak var downloadingLabels: UILabel!
+  @IBOutlet weak var downloadingLabelm: UILabel!
+  @IBOutlet weak var downloadingLabell: UILabel!
+  @IBOutlet weak var downloadingLabelx: UILabel!
+  @IBOutlet weak var priorizeButtonn: UIButton!
+  @IBOutlet weak var priorizeButtons: UIButton!
+  @IBOutlet weak var priorizeButtonm: UIButton!
+  @IBOutlet weak var priorizeButtonl: UIButton!
+  @IBOutlet weak var priorizeButtonx: UIButton!
+  let selection = UISelectionFeedbackGenerator()
+  var detector: VNCoreMLModel!
+  var session: AVCaptureSession!
+  var videoCapture: VideoCapture!
+  var currentBuffer: CVPixelBuffer?
+  var framesDone = 0
+  var t0 = 0.0  // inference start
+  var t1 = 0.0  // inference dt
+  var t2 = 0.0  // inference dt smoothed
+  var t3 = CACurrentMediaTime()  // FPS start
+  var t4 = 0.0  // FPS dt smoothed
+  // var cameraOutput: AVCapturePhotoOutput!
+
+  // Developer mode
+  let developerMode = UserDefaults.standard.bool(forKey: "developer_mode")  // developer mode selected in settings
+  let save_detections = false  // write every detection to detections.txt
+  let save_frames = false  // write every frame to frames.txt
+
+  lazy var visionRequest: VNCoreMLRequest = {
+    let request = VNCoreMLRequest(
+      model: detector,
+      completionHandler: {
+        [weak self] request, error in
+        self?.processObservations(for: request, error: error)
+      })
+    // NOTE: BoundingBoxView object scaling depends on request.imageCropAndScaleOption https://developer.apple.com/documentation/vision/vnimagecropandscaleoption
+    request.imageCropAndScaleOption = .scaleFill  // .scaleFit, .scaleFill, .centerCrop
+    return request
+  }()
+
+  var firstModelLoaded = false
+
+  override func viewDidLoad() {
+    super.viewDidLoad()
+    //        ModelFileManager.shared.deleteAllDownloadedModels()
+    //        ModelCacheManager.shared.loadBundledModel()
+    loadFirstModel()
+    slider.value = 30
+    setLabels()
+    setupSegmentecControl()
+    startVideo()
+    // setModel()
+  }
+
+  @IBAction func vibrate(_ sender: Any) {
+    selection.selectionChanged()
+  }
+
+  @IBAction func indexChanged(_ sender: Any) {
+    selection.selectionChanged()
+    activityIndicator.startAnimating()
+
+    /// Switch model
+    switch segmentedControl.selectedSegmentIndex {
+    case 0:
+      let modelName = "yolov8n"
+      self.labelName.text = "YOLOv8n"
+      userDidSelectFile(with: modelName)
+    case 1:
+      let modelName = "yolov8s"
+      self.labelName.text = "YOLOv8s"
+      userDidSelectFile(with: modelName)
+    case 2:
+      let modelName = "yolov8m"
+      self.labelName.text = "YOLOv8m"
+      userDidSelectFile(with: modelName)
+    case 3:
+      let modelName = "yolov8l"
+      self.labelName.text = "YOLOv8l"
+      userDidSelectFile(with: modelName)
+    case 4:
+      let modelName = "yolov8x"
+      self.labelName.text = "YOLOv8x"
+      userDidSelectFile(with: modelName)
+    default:
+      break
+    }
+    setModel()
+    setUpBoundingBoxViews()
+    activityIndicator.stopAnimating()
+  }
+
+  private func loadFirstModel() {
+    guard let firstModel = firstModels["detect"] else { fatalError("invalid model url") }
+    let key = firstModel.0
+    let remoteURL = firstModel.1
+    let fileName = remoteURL.lastPathComponent
+    ModelCacheManager.shared.loadModel(from: fileName, remoteURL: remoteURL, key: key) {
+      [self] model, key in
+      if let model = model {
+        segmentedControl.selectedSegmentIndex = 2
+        mlModel = model
+        detector = try! VNCoreMLModel(for: mlModel)
         setUpBoundingBoxViews()
-        startVideo()
-        // setModel()
+        // Add the bounding box layers to the UI, on top of the video preview.
+        for box in self.boundingBoxViews {
+          box.addToLayer(self.videoPreview.layer)
+        }
+        enableSegmentedControl(key: key)
+        firstModelLoaded = true
+        downloadAllModels()
+
+        print("Loaded model for key: \(key)")
+      } else {
+        print("Failed to load model for key: \(key)")
+      }
     }
-
-    @IBAction func vibrate(_ sender: Any) {
-        selection.selectionChanged()
+  }
+
+  private func downloadAllModels() {
+    for (key, remoteURL) in fileMappings {
+      let fileName = remoteURL.lastPathComponent
+      ModelCacheManager.shared.loadModel(from: fileName, remoteURL: remoteURL, key: key) {
+        [self] model, key in
+        if let model = model {
+          enableSegmentedControl(key: key)
+          print("Loaded model for key: \(key)")
+        } else {
+          print("Failed to load model for key: \(key)")
+        }
+      }
     }
+  }
+
+  func userDidSelectFile(with key: String) {
+    print(ModelCacheManager.shared.modelCache.keys)
+
+    //        if let currentSelectedKey = ModelCacheManager.shared.getCurrentSelectedModelKey() {
+    //            if currentSelectedKey != key {
+    //                ModelDownloadManager.shared.cancelCurrentDownload()
+    //            }
+    //        }
+
+    guard presetModels.contains(key) else { return }
+    let fileName = key
+
+    ModelCacheManager.shared.setCurrentSelectedModelKey(key)
+
+    if let cachedModel = ModelCacheManager.shared.modelCache[key] {
+      print("Model loaded from cache for key: \(key)")
+      enableSegmentedControl(key: key)
+      mlModel = cachedModel
+      setRequest()
+    } else if ModelCacheManager.shared.isModelDownloaded(key: key) {
+      let selectedURL = fileMappings.first { $0.0 == key }?.1
+      guard let remoteURL = selectedURL else {
+        print("Invalid key: \(key)")
+        return
+      }
+      ModelCacheManager.shared.loadModel(from: fileName, remoteURL: remoteURL, key: key) {
+        [self] model, key in
+        if let model = model {
+          print("Model loaded from local storage for key: \(key)")
+          enableSegmentedControl(key: key)
+          mlModel = model
+          setRequest()
 
-    @IBAction func indexChanged(_ sender: Any) {
-        selection.selectionChanged()
-        activityIndicator.startAnimating()
-
-        /// Switch model
-        switch segmentedControl.selectedSegmentIndex {
+        } else {
+          print("Failed to load model from local storage for key: \(key)")
+        }
+      }
+    } else {
+      ModelCacheManager.shared.prioritizeDownload(for: fileName) { [self] model, key in
+        if let model = model {
+          print("User selected model is ready for key: \(key)")
+          enableSegmentedControl(key: key)
+          //                    mlModel = model
+          //                    setRequest()
+        } else {
+          print("Failed to initialize model for key: \(key)")
+        }
+      }
+    }
+  }
+
+  func setModel() {
+    /// VNCoreMLModel
+    detector = try! VNCoreMLModel(for: mlModel)
+    detector.featureProvider = ThresholdProvider()
+
+    /// VNCoreMLRequest
+    let request = VNCoreMLRequest(
+      model: detector,
+      completionHandler: { [weak self] request, error in
+        self?.processObservations(for: request, error: error)
+      })
+    request.imageCropAndScaleOption = .scaleFill  // .scaleFit, .scaleFill, .centerCrop
+    visionRequest = request
+    t2 = 0.0  // inference dt smoothed
+    t3 = CACurrentMediaTime()  // FPS start
+    t4 = 0.0  // FPS dt smoothed
+  }
+
+  func setRequest() {
+    detector = try! VNCoreMLModel(for: mlModel)
+    detector.featureProvider = ThresholdProvider()
+
+    /// VNCoreMLRequest
+    let request = VNCoreMLRequest(
+      model: detector,
+      completionHandler: { [weak self] request, error in
+        self?.processObservations(for: request, error: error)
+      })
+    request.imageCropAndScaleOption = .scaleFill  // .scaleFit, .scaleFill, .centerCrop
+    visionRequest = request
+    t2 = 0.0  // inference dt smoothed
+    t3 = CACurrentMediaTime()  // FPS start
+    t4 = 0.0  // FPS dt smoothed
+  }
+
+  func setupSegmentecControl() {
+    priorizeButtonn.setTitle("", for: .normal)
+    priorizeButtons.setTitle("", for: .normal)
+    priorizeButtonm.setTitle("", for: .normal)
+    priorizeButtonl.setTitle("", for: .normal)
+    priorizeButtonx.setTitle("", for: .normal)
+
+    var enableIndexes: [Int] = []
+    for (i, modelName) in presetModels.enumerated() {
+      if ModelCacheManager.shared.isModelDownloaded(key: modelName) {
+        enableIndexes.append(i)
+      }
+    }
+    for i in 0...4 {
+      if !enableIndexes.contains(i) {
+        self.segmentedControl.setEnabled(false, forSegmentAt: i)
+        switch i {
         case 0:
-            self.labelName.text = "YOLOv8n"
-            mlModel = try! yolov8n(configuration: .init()).model
+          downloadingLabeln.isHidden = false
+          priorizeButtonn.isHidden = false
         case 1:
-            self.labelName.text = "YOLOv8s"
-            mlModel = try! yolov8s(configuration: .init()).model
+          downloadingLabels.isHidden = false
+          priorizeButtons.isHidden = false
         case 2:
-            self.labelName.text = "YOLOv8m"
-            mlModel = try! yolov8m(configuration: .init()).model
+          downloadingLabelm.isHidden = false
+          priorizeButtonm.isHidden = false
         case 3:
-            self.labelName.text = "YOLOv8l"
-            mlModel = try! yolov8l(configuration: .init()).model
+          downloadingLabell.isHidden = false
+          priorizeButtonl.isHidden = false
         case 4:
-            self.labelName.text = "YOLOv8x"
-            mlModel = try! yolov8x(configuration: .init()).model
+          downloadingLabelx.isHidden = false
+          priorizeButtonx.isHidden = false
         default:
-            break
+          break
         }
-        setModel()
-        setUpBoundingBoxViews()
-        activityIndicator.stopAnimating()
+      }
     }
-
-    func setModel() {
-        /// VNCoreMLModel
-        detector = try! VNCoreMLModel(for: mlModel)
-        detector.featureProvider = ThresholdProvider()
-
-        /// VNCoreMLRequest
-        let request = VNCoreMLRequest(model: detector, completionHandler: { [weak self] request, error in
-            self?.processObservations(for: request, error: error)
-        })
-        request.imageCropAndScaleOption = .scaleFill  // .scaleFit, .scaleFill, .centerCrop
-        visionRequest = request
-        t2 = 0.0 // inference dt smoothed
-        t3 = CACurrentMediaTime()  // FPS start
-        t4 = 0.0  // FPS dt smoothed
+  }
+
+  func enableSegmentedControl(key: String) {
+    guard let index = presetModels.firstIndex(of: key) else { return }
+    self.segmentedControl.setEnabled(true, forSegmentAt: index)
+    switch index {
+    case 0:
+      downloadingLabeln.isHidden = true
+      priorizeButtonn.isHidden = true
+    case 1:
+      downloadingLabels.isHidden = true
+      priorizeButtons.isHidden = true
+    case 2:
+      downloadingLabelm.isHidden = true
+      priorizeButtonm.isHidden = true
+    case 3:
+      downloadingLabell.isHidden = true
+      priorizeButtonl.isHidden = true
+    case 4:
+      downloadingLabelx.isHidden = true
+      priorizeButtonl.isHidden = true
+    default:
+      break
     }
-
-    /// Update thresholds from slider values
-    @IBAction func sliderChanged(_ sender: Any) {
-        let conf = Double(round(100 * sliderConf.value)) / 100
-        let iou = Double(round(100 * sliderIoU.value)) / 100
-        self.labelSliderConf.text = String(conf) + " Confidence Threshold"
-        self.labelSliderIoU.text = String(iou) + " IoU Threshold"
-        detector.featureProvider = ThresholdProvider(iouThreshold: iou, confidenceThreshold: conf)
+  }
+
+  private func priorizeDownload(key: String) {
+    ModelCacheManager.shared.prioritizeDownload(for: key) { [self] model, key in
+      if let model = model {
+        print("User selected model is ready for key: \(key)")
+        enableSegmentedControl(key: key)
+        //                mlModel = model
+        //                setRequest()
+      } else {
+        print("Failed to initialize model for key: \(key)")
+      }
     }
 
-    @IBAction func takePhoto(_ sender: Any?) {
-        let t0 = DispatchTime.now().uptimeNanoseconds
-
-        // 1. captureSession and cameraOutput
-        // session = videoCapture.captureSession  // session = AVCaptureSession()
-        // session.sessionPreset = AVCaptureSession.Preset.photo
-        // cameraOutput = AVCapturePhotoOutput()
-        // cameraOutput.isHighResolutionCaptureEnabled = true
-        // cameraOutput.isDualCameraDualPhotoDeliveryEnabled = true
-        // print("1 Done: ", Double(DispatchTime.now().uptimeNanoseconds - t0) / 1E9)
-
-        // 2. Settings
-        let settings = AVCapturePhotoSettings()
-        // settings.flashMode = .off
-        // settings.isHighResolutionPhotoEnabled = cameraOutput.isHighResolutionCaptureEnabled
-        // settings.isDualCameraDualPhotoDeliveryEnabled = self.videoCapture.cameraOutput.isDualCameraDualPhotoDeliveryEnabled
-
-        // 3. Capture Photo
-        usleep(20_000)  // short 10 ms delay to allow camera to focus
-        self.videoCapture.cameraOutput.capturePhoto(with: settings, delegate: self as AVCapturePhotoCaptureDelegate)
-        print("3 Done: ", Double(DispatchTime.now().uptimeNanoseconds - t0) / 1E9)
+  }
+
+  @IBAction func priorizeButtonnTapped(_ sender: UIButton) {
+    sender.isHidden = true
+    priorizeDownload(key: "yolov8n")
+    downloadingLabeln.text = "prioritized"
+    downloadingLabeln.textColor = .red
+  }
+
+  @IBAction func priorizeButtonsTapped(_ sender: UIButton) {
+    sender.isHidden = true
+    priorizeDownload(key: "yolov8s")
+    downloadingLabels.text = "prioritized"
+    downloadingLabels.textColor = .red
+  }
+  @IBAction func priorizeButtonmTapped(_ sender: UIButton) {
+    sender.isHidden = true
+    priorizeDownload(key: "yolov8m")
+    downloadingLabelm.text = "prioritized"
+    downloadingLabelm.textColor = .red
+  }
+
+  @IBAction func priorizeButtonlTapped(_ sender: UIButton) {
+    sender.isHidden = true
+    priorizeDownload(key: "yolov8l")
+    downloadingLabell.text = "prioritized"
+    downloadingLabell.textColor = .red
+  }
+  @IBAction func priorizeButtonxTapped(_ sender: UIButton) {
+    sender.isHidden = true
+    priorizeDownload(key: "yolov8x")
+    downloadingLabelx.text = "prioritized"
+    downloadingLabelx.textColor = .red
+  }
+  /// Update thresholds from slider values
+  @IBAction func sliderChanged(_ sender: Any) {
+    let conf = Double(round(100 * sliderConf.value)) / 100
+    let iou = Double(round(100 * sliderIoU.value)) / 100
+    self.labelSliderConf.text = String(conf) + " Confidence Threshold"
+    self.labelSliderIoU.text = String(iou) + " IoU Threshold"
+    detector.featureProvider = ThresholdProvider(iouThreshold: iou, confidenceThreshold: conf)
+  }
+
+  @IBAction func takePhoto(_ sender: Any?) {
+    let t0 = DispatchTime.now().uptimeNanoseconds
+
+    // 1. captureSession and cameraOutput
+    // session = videoCapture.captureSession  // session = AVCaptureSession()
+    // session.sessionPreset = AVCaptureSession.Preset.photo
+    // cameraOutput = AVCapturePhotoOutput()
+    // cameraOutput.isHighResolutionCaptureEnabled = true
+    // cameraOutput.isDualCameraDualPhotoDeliveryEnabled = true
+    // print("1 Done: ", Double(DispatchTime.now().uptimeNanoseconds - t0) / 1E9)
+
+    // 2. Settings
+    let settings = AVCapturePhotoSettings()
+    // settings.flashMode = .off
+    // settings.isHighResolutionPhotoEnabled = cameraOutput.isHighResolutionCaptureEnabled
+    // settings.isDualCameraDualPhotoDeliveryEnabled = self.videoCapture.cameraOutput.isDualCameraDualPhotoDeliveryEnabled
+
+    // 3. Capture Photo
+    usleep(20_000)  // short 10 ms delay to allow camera to focus
+    self.videoCapture.cameraOutput.capturePhoto(
+      with: settings, delegate: self as AVCapturePhotoCaptureDelegate)
+    print("3 Done: ", Double(DispatchTime.now().uptimeNanoseconds - t0) / 1E9)
+  }
+
+  @IBAction func logoButton(_ sender: Any) {
+    selection.selectionChanged()
+    if let link = URL(string: "https://www.ultralytics.com") {
+      UIApplication.shared.open(link)
     }
-
-    @IBAction func logoButton(_ sender: Any) {
-        selection.selectionChanged()
-        if let link = URL(string: "https://www.ultralytics.com") {
-            UIApplication.shared.open(link)
-        }
+  }
+
+  func setLabels() {
+    self.labelName.text = "YOLOv8m"
+    self.labelVersion.text = "Version " + UserDefaults.standard.string(forKey: "app_version")!
+  }
+
+  @IBAction func playButton(_ sender: Any) {
+    selection.selectionChanged()
+    self.videoCapture.start()
+    playButtonOutlet.isEnabled = false
+    pauseButtonOutlet.isEnabled = true
+  }
+
+  @IBAction func pauseButton(_ sender: Any?) {
+    selection.selectionChanged()
+    self.videoCapture.stop()
+    playButtonOutlet.isEnabled = true
+    pauseButtonOutlet.isEnabled = false
+  }
+
+  @IBAction func switchCameraTapped(_ sender: Any) {
+    self.videoCapture.captureSession.beginConfiguration()
+    let currentInput = self.videoCapture.captureSession.inputs.first as? AVCaptureDeviceInput
+    self.videoCapture.captureSession.removeInput(currentInput!)
+    // let newCameraDevice = currentInput?.device == .builtInWideAngleCamera ? getCamera(with: .front) : getCamera(with: .back)
+
+    let device = AVCaptureDevice.default(.builtInWideAngleCamera, for: .video, position: .back)!
+    guard let videoInput1 = try? AVCaptureDeviceInput(device: device) else {
+      return
     }
 
-    func setLabels() {
-        self.labelName.text = "YOLOv8m"
-        self.labelVersion.text = "Version " + UserDefaults.standard.string(forKey: "app_version")!
+    self.videoCapture.captureSession.addInput(videoInput1)
+    self.videoCapture.captureSession.commitConfiguration()
+  }
+
+  // share image
+  @IBAction func shareButton(_ sender: Any) {
+    selection.selectionChanged()
+    let settings = AVCapturePhotoSettings()
+    self.videoCapture.cameraOutput.capturePhoto(
+      with: settings, delegate: self as AVCapturePhotoCaptureDelegate)
+  }
+
+  // share screenshot
+  @IBAction func saveScreenshotButton(_ shouldSave: Bool = true) {
+    // let layer = UIApplication.shared.keyWindow!.layer
+    // let scale = UIScreen.main.scale
+    // UIGraphicsBeginImageContextWithOptions(layer.frame.size, false, scale);
+    // layer.render(in: UIGraphicsGetCurrentContext()!)
+    // let screenshot = UIGraphicsGetImageFromCurrentImageContext()
+    // UIGraphicsEndImageContext()
+
+    // let screenshot = UIApplication.shared.screenShot
+    // UIImageWriteToSavedPhotosAlbum(screenshot!, nil, nil, nil)
+  }
+
+  let maxBoundingBoxViews = 100
+  var boundingBoxViews = [BoundingBoxView]()
+  var colors: [String: UIColor] = [:]
+
+  func setUpBoundingBoxViews() {
+    // Ensure all bounding box views are initialized up to the maximum allowed.
+    while boundingBoxViews.count < maxBoundingBoxViews {
+      boundingBoxViews.append(BoundingBoxView())
     }
 
-    @IBAction func playButton(_ sender: Any) {
-        selection.selectionChanged()
-        self.videoCapture.start()
-        playButtonOutlet.isEnabled = false
-        pauseButtonOutlet.isEnabled = true
+    // Retrieve class labels directly from the CoreML model's class labels, if available.
+    guard let classLabels = mlModel.modelDescription.classLabels as? [String] else {
+      fatalError("Class labels are missing from the model description")
     }
 
-    @IBAction func pauseButton(_ sender: Any?) {
-        selection.selectionChanged()
-        self.videoCapture.stop()
-        playButtonOutlet.isEnabled = true
-        pauseButtonOutlet.isEnabled = false
+    // Assign random colors to the classes.
+    for label in classLabels {
+      if colors[label] == nil {  // if key not in dict
+        colors[label] = UIColor(
+          red: CGFloat.random(in: 0...1),
+          green: CGFloat.random(in: 0...1),
+          blue: CGFloat.random(in: 0...1),
+          alpha: 0.6)
+      }
     }
-
-    @IBAction func switchCameraTapped(_ sender: Any) {
-        self.videoCapture.captureSession.beginConfiguration()
-        let currentInput = self.videoCapture.captureSession.inputs.first as? AVCaptureDeviceInput
-        self.videoCapture.captureSession.removeInput(currentInput!)
-        // let newCameraDevice = currentInput?.device == .builtInWideAngleCamera ? getCamera(with: .front) : getCamera(with: .back)
-
-        let device = AVCaptureDevice.default(.builtInWideAngleCamera, for: .video, position: .back)!
-        guard let videoInput1 = try? AVCaptureDeviceInput(device: device) else {
-            return
+  }
+
+  func startVideo() {
+    videoCapture = VideoCapture()
+    videoCapture.delegate = self
+
+    videoCapture.setUp(sessionPreset: .photo) { success in
+      // .hd4K3840x2160 or .photo (4032x3024)  Warning: 4k may not work on all devices i.e. 2019 iPod
+      if success {
+        // Add the video preview into the UI.
+        if let previewLayer = self.videoCapture.previewLayer {
+          self.videoPreview.layer.addSublayer(previewLayer)
+          self.videoCapture.previewLayer?.frame = self.videoPreview.bounds  // resize preview layer
         }
 
-        self.videoCapture.captureSession.addInput(videoInput1)
-        self.videoCapture.captureSession.commitConfiguration()
+        // Once everything is set up, we can start capturing live video.
+        self.videoCapture.start()
+      }
     }
+  }
+
+  func predict(sampleBuffer: CMSampleBuffer) {
+
+    guard firstModelLoaded else { return }
+
+    if currentBuffer == nil, let pixelBuffer = CMSampleBufferGetImageBuffer(sampleBuffer) {
+      currentBuffer = pixelBuffer
+
+      /// - Tag: MappingOrientation
+      // The frame is always oriented based on the camera sensor,
+      // so in most cases Vision needs to rotate it for the model to work as expected.
+      let imageOrientation: CGImagePropertyOrientation
+      switch UIDevice.current.orientation {
+      case .portrait:
+        imageOrientation = .up
+      case .portraitUpsideDown:
+        imageOrientation = .down
+      case .landscapeLeft:
+        imageOrientation = .left
+      case .landscapeRight:
+        imageOrientation = .right
+      case .unknown:
+        print("The device orientation is unknown, the predictions may be affected")
+        fallthrough
+      default:
+        imageOrientation = .up
+      }
+
+      // Invoke a VNRequestHandler with that image
+      let handler = VNImageRequestHandler(
+        cvPixelBuffer: pixelBuffer, orientation: imageOrientation, options: [:])
+      if UIDevice.current.orientation != .faceUp {  // stop if placed down on a table
+        t0 = CACurrentMediaTime()  // inference start
+        do {
+          try handler.perform([visionRequest])
+        } catch {
+          print(error)
+        }
+        t1 = CACurrentMediaTime() - t0  // inference dt
+      }
 
-    // share image
-    @IBAction func shareButton(_ sender: Any) {
-        selection.selectionChanged()
-        let settings = AVCapturePhotoSettings()
-        self.videoCapture.cameraOutput.capturePhoto(with: settings, delegate: self as AVCapturePhotoCaptureDelegate)
+      currentBuffer = nil
     }
-
-    // share screenshot
-    @IBAction func saveScreenshotButton(_ shouldSave: Bool = true) {
-        // let layer = UIApplication.shared.keyWindow!.layer
-        // let scale = UIScreen.main.scale
-        // UIGraphicsBeginImageContextWithOptions(layer.frame.size, false, scale);
-        // layer.render(in: UIGraphicsGetCurrentContext()!)
-        // let screenshot = UIGraphicsGetImageFromCurrentImageContext()
-        // UIGraphicsEndImageContext()
-
-        // let screenshot = UIApplication.shared.screenShot
-        // UIImageWriteToSavedPhotosAlbum(screenshot!, nil, nil, nil)
+  }
+
+  func processObservations(for request: VNRequest, error: Error?) {
+    DispatchQueue.main.async {
+      if let results = request.results as? [VNRecognizedObjectObservation] {
+        self.show(predictions: results)
+      } else {
+        self.show(predictions: [])
+      }
+
+      // Measure FPS
+      if self.t1 < 10.0 {  // valid dt
+        self.t2 = self.t1 * 0.05 + self.t2 * 0.95  // smoothed inference time
+      }
+      self.t4 = (CACurrentMediaTime() - self.t3) * 0.05 + self.t4 * 0.95  // smoothed delivered FPS
+      self.labelFPS.text = String(format: "%.1f FPS - %.1f ms", 1 / self.t4, self.t2 * 1000)  // t2 seconds to ms
+      self.t3 = CACurrentMediaTime()
     }
-
-    let maxBoundingBoxViews = 100
-    var boundingBoxViews = [BoundingBoxView]()
-    var colors: [String: UIColor] = [:]
-
-    func setUpBoundingBoxViews() {
-        // Ensure all bounding box views are initialized up to the maximum allowed.
-        while boundingBoxViews.count < maxBoundingBoxViews {
-            boundingBoxViews.append(BoundingBoxView())
-        }
-
-        // Retrieve class labels directly from the CoreML model's class labels, if available.
-        guard let classLabels = mlModel.modelDescription.classLabels as? [String] else {
-            fatalError("Class labels are missing from the model description")
+  }
+
+  // Save text file
+  func saveText(text: String, file: String = "saved.txt") {
+    if let dir = FileManager.default.urls(for: .documentDirectory, in: .userDomainMask).first {
+      let fileURL = dir.appendingPathComponent(file)
+
+      // Writing
+      do {  // Append to file if it exists
+        let fileHandle = try FileHandle(forWritingTo: fileURL)
+        fileHandle.seekToEndOfFile()
+        fileHandle.write(text.data(using: .utf8)!)
+        fileHandle.closeFile()
+      } catch {  // Create new file and write
+        do {
+          try text.write(to: fileURL, atomically: false, encoding: .utf8)
+        } catch {
+          print("no file written")
         }
+      }
 
-        // Assign random colors to the classes.
-        for label in classLabels {
-            if colors[label] == nil {  // if key not in dict
-                colors[label] = UIColor(red: CGFloat.random(in: 0...1),
-                        green: CGFloat.random(in: 0...1),
-                        blue: CGFloat.random(in: 0...1),
-                        alpha: 0.6)
-            }
-        }
+      // Reading
+      // do {let text2 = try String(contentsOf: fileURL, encoding: .utf8)} catch {/* error handling here */}
     }
-
-    func startVideo() {
-        videoCapture = VideoCapture()
-        videoCapture.delegate = self
-
-        videoCapture.setUp(sessionPreset: .photo) { success in
-            // .hd4K3840x2160 or .photo (4032x3024)  Warning: 4k may not work on all devices i.e. 2019 iPod
-            if success {
-                // Add the video preview into the UI.
-                if let previewLayer = self.videoCapture.previewLayer {
-                    self.videoPreview.layer.addSublayer(previewLayer)
-                    self.videoCapture.previewLayer?.frame = self.videoPreview.bounds  // resize preview layer
-                }
-
-                // Add the bounding box layers to the UI, on top of the video preview.
-                for box in self.boundingBoxViews {
-                    box.addToLayer(self.videoPreview.layer)
-                }
-
-                // Once everything is set up, we can start capturing live video.
-                self.videoCapture.start()
-            }
-        }
+  }
+
+  // Save image file
+  func saveImage() {
+    let dir = FileManager.default.urls(for: .documentDirectory, in: .userDomainMask).first
+    let fileURL = dir!.appendingPathComponent("saved.jpg")
+    let image = UIImage(named: "ultralytics_yolo_logotype.png")
+    FileManager.default.createFile(
+      atPath: fileURL.path, contents: image!.jpegData(compressionQuality: 0.5), attributes: nil)
+  }
+
+  // Return hard drive space (GB)
+  func freeSpace() -> Double {
+    let fileURL = URL(fileURLWithPath: NSHomeDirectory() as String)
+    do {
+      let values = try fileURL.resourceValues(forKeys: [
+        .volumeAvailableCapacityForImportantUsageKey
+      ])
+      return Double(values.volumeAvailableCapacityForImportantUsage!) / 1E9  // Bytes to GB
+    } catch {
+      print("Error retrieving storage capacity: \(error.localizedDescription)")
     }
-
-    func predict(sampleBuffer: CMSampleBuffer) {
-        if currentBuffer == nil, let pixelBuffer = CMSampleBufferGetImageBuffer(sampleBuffer) {
-            currentBuffer = pixelBuffer
-
-            /// - Tag: MappingOrientation
-            // The frame is always oriented based on the camera sensor,
-            // so in most cases Vision needs to rotate it for the model to work as expected.
-            let imageOrientation: CGImagePropertyOrientation
-            switch UIDevice.current.orientation {
-            case .portrait:
-                imageOrientation = .up
-            case .portraitUpsideDown:
-                imageOrientation = .down
-            case .landscapeLeft:
-                imageOrientation = .left
-            case .landscapeRight:
-                imageOrientation = .right
-            case .unknown:
-                print("The device orientation is unknown, the predictions may be affected")
-                fallthrough
-            default:
-                imageOrientation = .up
-            }
-
-            // Invoke a VNRequestHandler with that image
-            let handler = VNImageRequestHandler(cvPixelBuffer: pixelBuffer, orientation: imageOrientation, options: [:])
-            if UIDevice.current.orientation != .faceUp {  // stop if placed down on a table
-                t0 = CACurrentMediaTime()  // inference start
-                do {
-                    try handler.perform([visionRequest])
-                } catch {
-                    print(error)
-                }
-                t1 = CACurrentMediaTime() - t0  // inference dt
-            }
-
-            currentBuffer = nil
-        }
+    return 0
+  }
+
+  // Return RAM usage (GB)
+  func memoryUsage() -> Double {
+    var taskInfo = mach_task_basic_info()
+    var count = mach_msg_type_number_t(MemoryLayout<mach_task_basic_info>.size) / 4
+    let kerr: kern_return_t = withUnsafeMutablePointer(to: &taskInfo) {
+      $0.withMemoryRebound(to: integer_t.self, capacity: 1) {
+        task_info(mach_task_self_, task_flavor_t(MACH_TASK_BASIC_INFO), $0, &count)
+      }
     }
-
-    func processObservations(for request: VNRequest, error: Error?) {
-        DispatchQueue.main.async {
-            if let results = request.results as? [VNRecognizedObjectObservation] {
-                self.show(predictions: results)
-            } else {
-                self.show(predictions: [])
-            }
-
-            // Measure FPS
-            if self.t1 < 10.0 {  // valid dt
-                self.t2 = self.t1 * 0.05 + self.t2 * 0.95  // smoothed inference time
-            }
-            self.t4 = (CACurrentMediaTime() - self.t3) * 0.05 + self.t4 * 0.95  // smoothed delivered FPS
-            self.labelFPS.text = String(format: "%.1f FPS - %.1f ms", 1 / self.t4, self.t2 * 1000)  // t2 seconds to ms
-            self.t3 = CACurrentMediaTime()
-        }
+    if kerr == KERN_SUCCESS {
+      return Double(taskInfo.resident_size) / 1E9  // Bytes to GB
+    } else {
+      return 0
     }
-
-    // Save text file
-    func saveText(text: String, file: String = "saved.txt") {
-        if let dir = FileManager.default.urls(for: .documentDirectory, in: .userDomainMask).first {
-            let fileURL = dir.appendingPathComponent(file)
-
-            // Writing
-            do {  // Append to file if it exists
-                let fileHandle = try FileHandle(forWritingTo: fileURL)
-                fileHandle.seekToEndOfFile()
-                fileHandle.write(text.data(using: .utf8)!)
-                fileHandle.closeFile()
-            } catch {  // Create new file and write
-                do {
-                    try text.write(to: fileURL, atomically: false, encoding: .utf8)
-                } catch {
-                    print("no file written")
-                }
-            }
-
-            // Reading
-            // do {let text2 = try String(contentsOf: fileURL, encoding: .utf8)} catch {/* error handling here */}
-        }
-    }
-
-    // Save image file
-    func saveImage() {
-        let dir = FileManager.default.urls(for: .documentDirectory, in: .userDomainMask).first
-        let fileURL = dir!.appendingPathComponent("saved.jpg")
-        let image = UIImage(named: "ultralytics_yolo_logotype.png")
-        FileManager.default.createFile(atPath: fileURL.path, contents: image!.jpegData(compressionQuality: 0.5), attributes: nil)
+  }
+
+  func show(predictions: [VNRecognizedObjectObservation]) {
+    let width = videoPreview.bounds.width  // 375 pix
+    let height = videoPreview.bounds.height  // 812 pix
+    var str = ""
+
+    // ratio = videoPreview AR divided by sessionPreset AR
+    var ratio: CGFloat = 1.0
+    if videoCapture.captureSession.sessionPreset == .photo {
+      ratio = (height / width) / (4.0 / 3.0)  // .photo
+    } else {
+      ratio = (height / width) / (16.0 / 9.0)  // .hd4K3840x2160, .hd1920x1080, .hd1280x720 etc.
     }
 
-    // Return hard drive space (GB)
-    func freeSpace() -> Double {
-        let fileURL = URL(fileURLWithPath: NSHomeDirectory() as String)
-        do {
-            let values = try fileURL.resourceValues(forKeys: [.volumeAvailableCapacityForImportantUsageKey])
-            return Double(values.volumeAvailableCapacityForImportantUsage!) / 1E9   // Bytes to GB
-        } catch {
-            print("Error retrieving storage capacity: \(error.localizedDescription)")
+    // date
+    let date = Date()
+    let calendar = Calendar.current
+    let hour = calendar.component(.hour, from: date)
+    let minutes = calendar.component(.minute, from: date)
+    let seconds = calendar.component(.second, from: date)
+    let nanoseconds = calendar.component(.nanosecond, from: date)
+    let sec_day =
+      Double(hour) * 3600.0 + Double(minutes) * 60.0 + Double(seconds) + Double(nanoseconds) / 1E9  // seconds in the day
+
+    self.labelSlider.text =
+      String(predictions.count) + " items (max " + String(Int(slider.value)) + ")"
+    for i in 0..<boundingBoxViews.count {
+      if i < predictions.count && i < Int(slider.value) {
+        let prediction = predictions[i]
+
+        var rect = prediction.boundingBox  // normalized xywh, origin lower left
+        switch UIDevice.current.orientation {
+        case .portraitUpsideDown:
+          rect = CGRect(
+            x: 1.0 - rect.origin.x - rect.width,
+            y: 1.0 - rect.origin.y - rect.height,
+            width: rect.width,
+            height: rect.height)
+        case .landscapeLeft:
+          rect = CGRect(
+            x: rect.origin.y,
+            y: 1.0 - rect.origin.x - rect.width,
+            width: rect.height,
+            height: rect.width)
+        case .landscapeRight:
+          rect = CGRect(
+            x: 1.0 - rect.origin.y - rect.height,
+            y: rect.origin.x,
+            width: rect.height,
+            height: rect.width)
+        case .unknown:
+          print("The device orientation is unknown, the predictions may be affected")
+          fallthrough
+        default: break
         }
-        return 0
-    }
 
-    // Return RAM usage (GB)
-    func memoryUsage() -> Double {
-        var taskInfo = mach_task_basic_info()
-        var count = mach_msg_type_number_t(MemoryLayout<mach_task_basic_info>.size) / 4
-        let kerr: kern_return_t = withUnsafeMutablePointer(to: &taskInfo) {
-            $0.withMemoryRebound(to: integer_t.self, capacity: 1) {
-                task_info(mach_task_self_, task_flavor_t(MACH_TASK_BASIC_INFO), $0, &count)
-            }
+        if ratio >= 1 {  // iPhone ratio = 1.218
+          let offset = (1 - ratio) * (0.5 - rect.minX)
+          let transform = CGAffineTransform(scaleX: 1, y: -1).translatedBy(x: offset, y: -1)
+          rect = rect.applying(transform)
+          rect.size.width *= ratio
+        } else {  // iPad ratio = 0.75
+          let offset = (ratio - 1) * (0.5 - rect.maxY)
+          let transform = CGAffineTransform(scaleX: 1, y: -1).translatedBy(x: 0, y: offset - 1)
+          rect = rect.applying(transform)
+          rect.size.height /= ratio
         }
-        if kerr == KERN_SUCCESS {
-            return Double(taskInfo.resident_size) / 1E9   // Bytes to GB
-        } else {
-            return 0
-        }
-    }
 
-    func show(predictions: [VNRecognizedObjectObservation]) {
-        let width = videoPreview.bounds.width  // 375 pix
-        let height = videoPreview.bounds.height  // 812 pix
-        var str = ""
+        // Scale normalized to pixels [375, 812] [width, height]
+        rect = VNImageRectForNormalizedRect(rect, Int(width), Int(height))
 
-        // ratio = videoPreview AR divided by sessionPreset AR
-        var ratio: CGFloat = 1.0
-        if videoCapture.captureSession.sessionPreset == .photo {
-            ratio = (height / width) / (4.0 / 3.0)  // .photo
-        } else {
-            ratio = (height / width) / (16.0 / 9.0)  // .hd4K3840x2160, .hd1920x1080, .hd1280x720 etc.
-        }
+        // The labels array is a list of VNClassificationObservation objects,
+        // with the highest scoring class first in the list.
+        let bestClass = prediction.labels[0].identifier
+        let confidence = prediction.labels[0].confidence
+        // print(confidence, rect)  // debug (confidence, xywh) with xywh origin top left (pixels)
 
-        // date
-        let date = Date()
-        let calendar = Calendar.current
-        let hour = calendar.component(.hour, from: date)
-        let minutes = calendar.component(.minute, from: date)
-        let seconds = calendar.component(.second, from: date)
-        let nanoseconds = calendar.component(.nanosecond, from: date)
-        let sec_day = Double(hour) * 3600.0 + Double(minutes) * 60.0 + Double(seconds) + Double(nanoseconds) / 1E9  // seconds in the day
-
-        self.labelSlider.text = String(predictions.count) + " items (max " + String(Int(slider.value)) + ")"
-        for i in 0..<boundingBoxViews.count {
-            if i < predictions.count && i < Int(slider.value) {
-                let prediction = predictions[i]
-
-                var rect = prediction.boundingBox  // normalized xywh, origin lower left
-                switch UIDevice.current.orientation {
-                case .portraitUpsideDown:
-                    rect = CGRect(x: 1.0 - rect.origin.x - rect.width,
-                            y: 1.0 - rect.origin.y - rect.height,
-                            width: rect.width,
-                            height: rect.height)
-                case .landscapeLeft:
-                    rect = CGRect(x: rect.origin.y,
-                            y: 1.0 - rect.origin.x - rect.width,
-                            width: rect.height,
-                            height: rect.width)
-                case .landscapeRight:
-                    rect = CGRect(x: 1.0 - rect.origin.y - rect.height,
-                            y: rect.origin.x,
-                            width: rect.height,
-                            height: rect.width)
-                case .unknown:
-                    print("The device orientation is unknown, the predictions may be affected")
-                    fallthrough
-                default: break
-                }
-
-                if ratio >= 1 { // iPhone ratio = 1.218
-                    let offset = (1 - ratio) * (0.5 - rect.minX)
-                    let transform = CGAffineTransform(scaleX: 1, y: -1).translatedBy(x: offset, y: -1)
-                    rect = rect.applying(transform)
-                    rect.size.width *= ratio
-                } else { // iPad ratio = 0.75
-                    let offset = (ratio - 1) * (0.5 - rect.maxY)
-                    let transform = CGAffineTransform(scaleX: 1, y: -1).translatedBy(x: 0, y: offset - 1)
-                    rect = rect.applying(transform)
-                    rect.size.height /= ratio
-                }
-
-                // Scale normalized to pixels [375, 812] [width, height]
-                rect = VNImageRectForNormalizedRect(rect, Int(width), Int(height))
-
-                // The labels array is a list of VNClassificationObservation objects,
-                // with the highest scoring class first in the list.
-                let bestClass = prediction.labels[0].identifier
-                let confidence = prediction.labels[0].confidence
-                // print(confidence, rect)  // debug (confidence, xywh) with xywh origin top left (pixels)
-
-                // Show the bounding box.
-                boundingBoxViews[i].show(frame: rect,
-                        label: String(format: "%@ %.1f", bestClass, confidence * 100),
-                        color: colors[bestClass] ?? UIColor.white,
-                        alpha: CGFloat((confidence - 0.2) / (1.0 - 0.2) * 0.9))  // alpha 0 (transparent) to 1 (opaque) for conf threshold 0.2 to 1.0)
-
-                if developerMode {
-                    // Write
-                    if save_detections {
-                        str += String(format: "%.3f %.3f %.3f %@ %.2f %.1f %.1f %.1f %.1f\n",
-                                sec_day, freeSpace(), UIDevice.current.batteryLevel, bestClass, confidence,
-                                rect.origin.x, rect.origin.y, rect.size.width, rect.size.height)
-                    }
-
-                    // Action trigger upon detection
-                    // if false {
-                    //     if (bestClass == "car") {  // "cell phone", "car", "person"
-                    //         self.takePhoto(nil)
-                    //         // self.pauseButton(nil)
-                    //         sleep(2)
-                    //     }
-                    // }
-                }
-            } else {
-                boundingBoxViews[i].hide()
-            }
-        }
+        // Show the bounding box.
+        boundingBoxViews[i].show(
+          frame: rect,
+          label: String(format: "%@ %.1f", bestClass, confidence * 100),
+          color: colors[bestClass] ?? UIColor.white,
+          alpha: CGFloat((confidence - 0.2) / (1.0 - 0.2) * 0.9))  // alpha 0 (transparent) to 1 (opaque) for conf threshold 0.2 to 1.0)
 
-        // Write
         if developerMode {
-            if save_detections {
-                saveText(text: str, file: "detections.txt")  // Write stats for each detection
-            }
-            if save_frames {
-                str = String(format: "%.3f %.3f %.3f %.3f %.1f %.1f %.1f\n",
-                        sec_day, freeSpace(), memoryUsage(), UIDevice.current.batteryLevel,
-                        self.t1 * 1000, self.t2 * 1000, 1 / self.t4)
-                saveText(text: str, file: "frames.txt")  // Write stats for each image
-            }
+          // Write
+          if save_detections {
+            str += String(
+              format: "%.3f %.3f %.3f %@ %.2f %.1f %.1f %.1f %.1f\n",
+              sec_day, freeSpace(), UIDevice.current.batteryLevel, bestClass, confidence,
+              rect.origin.x, rect.origin.y, rect.size.width, rect.size.height)
+          }
+
+          // Action trigger upon detection
+          // if false {
+          //     if (bestClass == "car") {  // "cell phone", "car", "person"
+          //         self.takePhoto(nil)
+          //         // self.pauseButton(nil)
+          //         sleep(2)
+          //     }
+          // }
         }
+      } else {
+        boundingBoxViews[i].hide()
+      }
+    }
 
-        // Debug
-        // print(str)
-        // print(UIDevice.current.identifierForVendor!)
-        // saveImage()
+    // Write
+    if developerMode {
+      if save_detections {
+        saveText(text: str, file: "detections.txt")  // Write stats for each detection
+      }
+      if save_frames {
+        str = String(
+          format: "%.3f %.3f %.3f %.3f %.1f %.1f %.1f\n",
+          sec_day, freeSpace(), memoryUsage(), UIDevice.current.batteryLevel,
+          self.t1 * 1000, self.t2 * 1000, 1 / self.t4)
+        saveText(text: str, file: "frames.txt")  // Write stats for each image
+      }
     }
 
-    // Pinch to Zoom Start ---------------------------------------------------------------------------------------------
-    let minimumZoom: CGFloat = 1.0
-    let maximumZoom: CGFloat = 10.0
-    var lastZoomFactor: CGFloat = 1.0
+    // Debug
+    // print(str)
+    // print(UIDevice.current.identifierForVendor!)
+    // saveImage()
+  }
 
-    @IBAction func pinch(_ pinch: UIPinchGestureRecognizer) {
-        let device = videoCapture.captureDevice
+  // Pinch to Zoom Start ---------------------------------------------------------------------------------------------
+  let minimumZoom: CGFloat = 1.0
+  let maximumZoom: CGFloat = 10.0
+  var lastZoomFactor: CGFloat = 1.0
 
-        // Return zoom value between the minimum and maximum zoom values
-        func minMaxZoom(_ factor: CGFloat) -> CGFloat {
-            return min(min(max(factor, minimumZoom), maximumZoom), device.activeFormat.videoMaxZoomFactor)
-        }
+  @IBAction func pinch(_ pinch: UIPinchGestureRecognizer) {
+    let device = videoCapture.captureDevice
 
-        func update(scale factor: CGFloat) {
-            do {
-                try device.lockForConfiguration()
-                defer {
-                    device.unlockForConfiguration()
-                }
-                device.videoZoomFactor = factor
-            } catch {
-                print("\(error.localizedDescription)")
-            }
-        }
+    // Return zoom value between the minimum and maximum zoom values
+    func minMaxZoom(_ factor: CGFloat) -> CGFloat {
+      return min(min(max(factor, minimumZoom), maximumZoom), device.activeFormat.videoMaxZoomFactor)
+    }
 
-        let newScaleFactor = minMaxZoom(pinch.scale * lastZoomFactor)
-        switch pinch.state {
-        case .began: fallthrough
-        case .changed:
-            update(scale: newScaleFactor)
-            self.labelZoom.text = String(format: "%.2fx", newScaleFactor)
-            self.labelZoom.font = UIFont.preferredFont(forTextStyle: .title2)
-        case .ended:
-            lastZoomFactor = minMaxZoom(newScaleFactor)
-            update(scale: lastZoomFactor)
-            self.labelZoom.font = UIFont.preferredFont(forTextStyle: .body)
-        default: break
+    func update(scale factor: CGFloat) {
+      do {
+        try device.lockForConfiguration()
+        defer {
+          device.unlockForConfiguration()
         }
-    }  // Pinch to Zoom Start ------------------------------------------------------------------------------------------
+        device.videoZoomFactor = factor
+      } catch {
+        print("\(error.localizedDescription)")
+      }
+    }
+
+    let newScaleFactor = minMaxZoom(pinch.scale * lastZoomFactor)
+    switch pinch.state {
+    case .began, .changed:
+      update(scale: newScaleFactor)
+      self.labelZoom.text = String(format: "%.2fx", newScaleFactor)
+      self.labelZoom.font = UIFont.preferredFont(forTextStyle: .title2)
+    case .ended:
+      lastZoomFactor = minMaxZoom(newScaleFactor)
+      update(scale: lastZoomFactor)
+      self.labelZoom.font = UIFont.preferredFont(forTextStyle: .body)
+    default: break
+    }
+  }  // Pinch to Zoom Start ------------------------------------------------------------------------------------------
 }  // ViewController class End
 
 extension ViewController: VideoCaptureDelegate {
-    func videoCapture(_ capture: VideoCapture, didCaptureVideoFrame sampleBuffer: CMSampleBuffer) {
-        predict(sampleBuffer: sampleBuffer)
-    }
+  func videoCapture(_ capture: VideoCapture, didCaptureVideoFrame sampleBuffer: CMSampleBuffer) {
+    predict(sampleBuffer: sampleBuffer)
+  }
 }
 
 // Programmatically save image
 extension ViewController: AVCapturePhotoCaptureDelegate {
-    func photoOutput(_ output: AVCapturePhotoOutput, didFinishProcessingPhoto photo: AVCapturePhoto, error: Error?) {
-        if let error = error {
-            print("error occurred : \(error.localizedDescription)")
-        }
-        if let dataImage = photo.fileDataRepresentation() {
-            let dataProvider = CGDataProvider(data: dataImage as CFData)
-            let cgImageRef: CGImage! = CGImage(jpegDataProviderSource: dataProvider!, decode: nil, shouldInterpolate: true, intent: .defaultIntent)
-            let image = UIImage(cgImage: cgImageRef, scale: 0.5, orientation: UIImage.Orientation.right)
-            let imageView = UIImageView(image: image)
-            imageView.contentMode = .scaleAspectFill
-            imageView.frame = videoPreview.frame
-            let imageLayer = imageView.layer
-            var sublayers = videoPreview.layer.sublayers ?? []
-            let insertIndex = max(sublayers.count - 1, 0)
-            videoPreview.layer.insertSublayer(imageLayer, above: videoCapture.previewLayer)
-            
-            let bounds = UIScreen.main.bounds
-            UIGraphicsBeginImageContextWithOptions(bounds.size, true, 0.0)
-            self.View0.drawHierarchy(in: bounds, afterScreenUpdates: true)
-            let img = UIGraphicsGetImageFromCurrentImageContext()
-            UIGraphicsEndImageContext()
-            imageLayer.removeFromSuperlayer()
-            let activityViewController = UIActivityViewController(activityItems: [img!], applicationActivities: nil)
-            activityViewController.popoverPresentationController?.sourceView = self.View0
-            self.present(activityViewController, animated: true, completion: nil)
-//
-//            // Save to camera roll
-//            UIImageWriteToSavedPhotosAlbum(img!, nil, nil, nil);
-        } else {
-            print("AVCapturePhotoCaptureDelegate Error")
-        }
+  func photoOutput(
+    _ output: AVCapturePhotoOutput, didFinishProcessingPhoto photo: AVCapturePhoto, error: Error?
+  ) {
+    if let error = error {
+      print("error occurred : \(error.localizedDescription)")
     }
+    if let dataImage = photo.fileDataRepresentation() {
+      let dataProvider = CGDataProvider(data: dataImage as CFData)
+      let cgImageRef: CGImage! = CGImage(
+        jpegDataProviderSource: dataProvider!, decode: nil, shouldInterpolate: true,
+        intent: .defaultIntent)
+      let image = UIImage(cgImage: cgImageRef, scale: 0.5, orientation: UIImage.Orientation.right)
+      let imageView = UIImageView(image: image)
+      imageView.contentMode = .scaleAspectFill
+      imageView.frame = videoPreview.frame
+      let imageLayer = imageView.layer
+      var sublayers = videoPreview.layer.sublayers ?? []
+      let insertIndex = max(sublayers.count - 1, 0)
+      videoPreview.layer.insertSublayer(imageLayer, above: videoCapture.previewLayer)
+
+      let bounds = UIScreen.main.bounds
+      UIGraphicsBeginImageContextWithOptions(bounds.size, true, 0.0)
+      self.View0.drawHierarchy(in: bounds, afterScreenUpdates: true)
+      let img = UIGraphicsGetImageFromCurrentImageContext()
+      UIGraphicsEndImageContext()
+      imageLayer.removeFromSuperlayer()
+      let activityViewController = UIActivityViewController(
+        activityItems: [img!], applicationActivities: nil)
+      activityViewController.popoverPresentationController?.sourceView = self.View0
+      self.present(activityViewController, animated: true, completion: nil)
+      //
+      //            // Save to camera roll
+      //            UIImageWriteToSavedPhotosAlbum(img!, nil, nil, nil);
+    } else {
+      print("AVCapturePhotoCaptureDelegate Error")
+    }
+  }
 }
-
-
-