diff --git a/requirements.txt b/requirements.txt
index 8e47744..6b388e0 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,4 +2,5 @@ torch
 aiia
 pillow
 torchvision
-sklearn
\ No newline at end of file
+sklearn
+https://gitea.fabelous.app/Machine-Learning/AIIA.git
\ No newline at end of file