diff --git a/server/website/requirements.txt b/server/website/requirements.txt
index 0ff00b6..148b6df 100644
--- a/server/website/requirements.txt
+++ b/server/website/requirements.txt
@@ -19,3 +19,5 @@ scikit-learn==0.19.1
 scipy==1.0.0
 tensorflow==1.6.0
 threadpool==1.3.2
+torch==1.2.0
+torchvision==0.4.0