From 0e7846e6e78c8621e94abe45fbfbdb8698684c29 Mon Sep 17 00:00:00 2001
From: Sun Jin Kim <sk2521@ic.ac.uk>
Date: Thu, 21 Apr 2022 19:19:54 +0100
Subject: [PATCH] edit gitlab yml

---
 docs/requirements.txt | 3 ++-
 requirements.txt      | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/docs/requirements.txt b/docs/requirements.txt
index 19c357e1..817a7012 100644
--- a/docs/requirements.txt
+++ b/docs/requirements.txt
@@ -1,3 +1,4 @@
-pytorch
+torch
 torchvision
+numpy
 matplotlib
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
index acf39854..d4be33f6 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,6 @@
-pytorch
+torch
 torchvision
+numpy
 Flask
 react
 pytest
-- 
GitLab