diff --git a/download.bash b/download.bash
new file mode 100755
index 0000000..5fcc9c7
--- /dev/null
+++ b/download.bash
@@ -0,0 +1,11 @@
+#!/bin/bash
+if [ ! -f craft_mlt_25k.pth ];then
+    gdown https://drive.google.com/uc?id=1Jk4eGD7crsqCCg9C9VjCLkMN3ze8kutZ
+fi
+if [ ! -f craft_ic15_20k.pth ];then
+    gdown https://drive.google.com/uc?id=1i2R7UIUqmkUtF0jv_3MXTqmQ_9wuAnLf
+fi
+if [ ! -f craft_refiner_CTW1500.pth ];then
+    gdown https://drive.google.com/uc?id=1XSaFwBkOaFOdtk4Ane3DFyJGPRw6v5bO
+fi
+
diff --git a/requirements.txt b/requirements.txt
index f4b2412..c71a6f5 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,7 @@
-torch==0.4.1.post2
+torch==0.4.1
 torchvision==0.2.1
 opencv-python==3.4.2.17
 scikit-image==0.14.2
-scipy==1.1.0
\ No newline at end of file
+scipy==1.1.0
+Pillow=6.2.1
+