diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 4386ca50ebfd9abf5adf653ab4af57c1ae9c7ffe..24467312375398074feedac862ae80548b3db998 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -9,12 +9,14 @@ stages:
 build:
   stage: build
   script:
+    - pip install -r requirements.txt
     - python machine_learning.py
 
 # Test stage
 test:
   stage: test
   script:
+    - pip install -r requirements.txt
     - pytest test_prediction.py
 
 # Export stage
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000000000000000000000000000000000000..898eb30702193e8277daabb1fea33096e439f8df
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,5 @@
+numpy
+matplotlib
+scikit-learn
+pytest
+joblib
\ No newline at end of file