diff --git a/requirements.txt b/requirements.txt
index 8eccbc3..b311546 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,5 @@
 tf-keras
+huggingface-hub
 transformers
 scikit-learn
 pandas
diff --git a/setup.py b/setup.py
index 2742bab..973c177 100755
--- a/setup.py
+++ b/setup.py
@@ -31,6 +31,7 @@
     python_requires=">=3.6",
     install_requires=[
         'tf-keras',
+        'huggingface-hub',
         'transformers',
         'scikit-learn',
         'pandas',