diff --git a/.ipynb_checkpoints/main-checkpoint.py b/.ipynb_checkpoints/main-checkpoint.py
index 9255423..e946d1d 100644
--- a/.ipynb_checkpoints/main-checkpoint.py
+++ b/.ipynb_checkpoints/main-checkpoint.py
@@ -33,9 +33,9 @@
     # paste your code here
     # path = os.getcwd()  # 获取当前工作目录
     # print(path)
-    if conf['style'][0] == '藏头诗':
+    if conf['style'] == ['藏头诗']:
         poetry = writer.cangtou(conf['Chinese_word'])
-    elif conf['style'][0] == '藏字诗':
+    elif conf['style'] == ['藏字诗']:
         poetry = writer.hide_words(conf['Chinese_word'])
     else:
         poetry = writer.rhyme_verse()
diff --git a/main.py b/main.py
index 9255423..e946d1d 100644
--- a/main.py
+++ b/main.py
@@ -33,9 +33,9 @@
     # paste your code here
     # path = os.getcwd()  # 获取当前工作目录
     # print(path)
-    if conf['style'][0] == '藏头诗':
+    if conf['style'] == ['藏头诗']:
         poetry = writer.cangtou(conf['Chinese_word'])
-    elif conf['style'][0] == '藏字诗':
+    elif conf['style'] == ['藏字诗']:
         poetry = writer.hide_words(conf['Chinese_word'])
     else:
         poetry = writer.rhyme_verse()
diff --git a/project_requirements.txt b/project_requirements.txt
new file mode 100644
index 0000000..e88b286
--- /dev/null
+++ b/project_requirements.txt
@@ -0,0 +1,19 @@
+torchvision==0.2.1
+toolz==0.9.0
+cloudpickle==0.6.1
+Cython==0.29
+pydot==1.2.4
+dask==0.20.2
+scikit-image==0.14.1
+networkx==2.2
+word2vec==0.10.2
+xlrd==1.1.0
+jieba==0.39
+defusedxml==0.5.0
+imgaug==0.2.6
+Shapely==1.6.4.post2
+nltk==3.3
+torch==0.4.1
+PyWavelets==1.0.1
+imageio==2.4.1
+python-jsonrpc-server==0.0.1