--- a
+++ b/requirments.txt
@@ -0,0 +1,5 @@
+numpy==1.21.0
+scikit-image==0.18.2
+scikit-learn==0.24.2
+scipy==1.7.0
+torch==1.9.0