diff --git a/aimodel/requirements.txt b/aimodel/requirements.txt
index 4830ce5..31472d1 100644
--- a/aimodel/requirements.txt
+++ b/aimodel/requirements.txt
@@ -1,3 +1,4 @@
 tensorflow>=2.4
+tensorflow_addons
 silence_tensorflow
 loguru
\ No newline at end of file