diff --git a/.github/workflows/unit-test.yml b/.github/workflows/unit-test.yml
index 44047a9..005643a 100644
--- a/.github/workflows/unit-test.yml
+++ b/.github/workflows/unit-test.yml
@@ -20,7 +20,7 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          pip install tensorflow==2.12.0 efficientnet==1.1.1 tflite-runtime==2.14.0
+          pip install tensorflow==2.12.0 efficientnet==1.1.1
           pip install -e .
 
       - name: Run tests
diff --git a/API/requirements.txt b/API/requirements.txt
index eb32831..0de96ea 100644
--- a/API/requirements.txt
+++ b/API/requirements.txt
@@ -1,6 +1,6 @@
 fastapi==0.103.2
 uvicorn==0.22.0
-fairpredictor==0.0.34
-tflite-runtime==2.14.0
+fairpredictor
+# tflite-runtime==2.14.0
 tensorflow==2.12.0 
 efficientnet==1.1.1
\ No newline at end of file