argument error. fix #158

Author: carpedm20

Diff for: README.md

@@ -29,7 +29,7 @@ Tensorflow implementation of [Deep Convolutional Generative Adversarial Networks
 
 First, download dataset with:
 
-    $ python download.py --datasets mnist celebA
+    $ python download.py mnist celebA
 
 To train a model with downloaded dataset:
 

Diff for: download.py

@@ -172,7 +172,7 @@ def prepare_data_dir(path = './data'):
   args = parser.parse_args()
   prepare_data_dir()
 
-  if 'celebA' in args.datasets:
+  if any(name in args.datasets for name in ['CelebA', 'celebA', 'celebA']):
     download_celeb_a('./data')
   if 'lsun' in args.datasets:
     download_lsun('./data')