diff --git a/requirements-huggingface.txt b/requirements-huggingface.txt
index c3e474f..11d8bbb 100644
--- a/requirements-huggingface.txt
+++ b/requirements-huggingface.txt
@@ -1,7 +1,7 @@
 tensorflow==2.15.0
 tensorflow_io
 tensorflow-addons
-tensorflow-probability==0.15.0
+tensorflow-probability
 numpy
 opencv-python
 # for Huggingface