diff --git a/gradle/libs.versions.toml b/gradle/libs.versions.toml
index 21201832..56ed2732 100644
--- a/gradle/libs.versions.toml
+++ b/gradle/libs.versions.toml
@@ -20,12 +20,11 @@ kover = "0.9.1"
 ktlint = "11.0.0"
 licensee = "1.7.0"
 lifecycle = "2.4.0"
+litert = "1.4.0"
 mockk = "1.13.4"
 robolectric = "4.14.1"
 roborazzi = "1.43.1"
 serialization = "1.3.3"
-tensorflow = "2.0.0"
-tensorflow-support = "0.3.0"
 turbine = "1.0.0"
 zxing = "3.5.2"
 
@@ -64,8 +63,8 @@ kotlin-serialization-json = { module = "org.jetbrains.kotlinx:kotlinx-serializat
 kotlin-coroutines = { module = "org.jetbrains.kotlinx:kotlinx-coroutines-android", version.ref = "coroutines" }
 
 # TensorFlow
-tensorflow = { module = "org.tensorflow:tensorflow-lite", version.ref = "tensorflow" }
-tensorflow-support = {module = "org.tensorflow:tensorflow-lite-support", version.ref = "tensorflow-support" }
+litert = { module = "com.google.ai.edge.litert:litert", version.ref = "litert" }
+litert-support = { module = "com.google.ai.edge.litert:litert-support", version.ref = "litert" }
 
 # Other
 zxing = { module = "com.google.zxing:core", version.ref = "zxing" }
diff --git a/liveness/build.gradle.kts b/liveness/build.gradle.kts
index 0e844e76..cc367574 100644
--- a/liveness/build.gradle.kts
+++ b/liveness/build.gradle.kts
@@ -9,7 +9,7 @@ android {
         externalNativeBuild {
             cmake {
                 cppFlags += "-std=c++17"
-                arguments += "-DCMAKE_VERBOSE_MAKEFILE=ON"
+                arguments += listOf("-DCMAKE_VERBOSE_MAKEFILE=ON", "-DANDROID_SUPPORT_FLEXIBLE_PAGE_SIZES=ON")
             }
         }
         buildConfigField(
@@ -58,8 +58,8 @@ dependencies {
 
     implementation(libs.kotlin.serialization.json)
 
-    implementation(libs.tensorflow)
-    implementation(libs.tensorflow.support)
+    implementation(libs.litert)
+    implementation(libs.litert.support)
 
     coreLibraryDesugaring(libs.android.desugar)