diff --git a/android/build.gradle b/android/build.gradle
index 57f0428..9538464 100644
--- a/android/build.gradle
+++ b/android/build.gradle
@@ -63,8 +63,8 @@ android {
 
 
 dependencies {
-    def tflite_version = "2.12.0"
+    def tflite_version = "1.4.0"
 
-    implementation("org.tensorflow:tensorflow-lite:${tflite_version}")
-    implementation("org.tensorflow:tensorflow-lite-gpu:${tflite_version}")
+    implementation("com.google.ai.edge.litert:litert:${tflite_version}")
+    implementation("com.google.ai.edge.litert:litert-gpu-api:${tflite_version}")
 }