diff --git a/ppgan/apps/psgan_predictor.py b/ppgan/apps/psgan_predictor.py
index 46b59220ae5146786a73635ed2cb3f07e2ca432c..6bc58c4cb654b2c4bc199c40b6babd3a39ab85de 100644
--- a/ppgan/apps/psgan_predictor.py
+++ b/ppgan/apps/psgan_predictor.py
@@ -185,14 +185,22 @@ class PSGANPredictor(BasePredictor):
         inference = Inference(self.cfg, self.weight_path)
         postprocess = PostProcess(self.cfg)
 
-        source = Image.open(self.args.source_path).convert("RGB")
+        try:
+            source = Image.open(self.args.source_path).convert("RGB")
+        except IOError:
+            print("Error: {} is not exist".format(self.args.source_path))
+            sys.exit()
+
         reference_paths = list(Path(self.args.reference_dir).glob("*"))
+        if len(reference_paths) == 0:
+            print("Error: Can't find image file in {}.".format(
+                self.args.reference_dir))
+            sys.exit()
         np.random.shuffle(reference_paths)
         for reference_path in reference_paths:
             if not reference_path.is_file():
                 print(reference_path, "is not a valid file.")
                 continue
-
             reference = Image.open(reference_path).convert("RGB")
 
             # Transfer the psgan from reference to source.
diff --git a/requirments.txt b/requirments.txt
index 2816f6be1834c8b8dc10b9cd0a5de48ba1f9614c..6f6de9a77f009d0911ea1699d15ad6a4bf9852f5 100644
--- a/requirments.txt
+++ b/requirments.txt
@@ -4,3 +4,4 @@ scikit-image>=0.14.0
 scipy>=1.1.0
 opencv-python
 imageio-ffmpeg
+dlib
diff --git a/setup.py b/setup.py
index 004f9fa098a2fbdce7bb54d751dfe7d8a3ed3e9f..54c197caa33d24ebc003bff793fc56e0253186f8 100644
--- a/setup.py
+++ b/setup.py
@@ -20,7 +20,7 @@ with open('requirments.txt', encoding="utf-8-sig") as f:
 
 
 def readme():
-    with open('doc/doc_en/whl_en.md', encoding="utf-8-sig") as f:
+    with open('README_en.md', encoding="utf-8-sig") as f:
         README = f.read()
     return README
 
@@ -30,10 +30,13 @@ setup(
     packages=['ppgan'],
     include_package_data=True,
     entry_points={"console_scripts": ["paddlegan= paddlegan.paddlegan:main"]},
+    author='PaddlePaddle Author',
     version='0.1.0',
     install_requires=requirements,
     license='Apache License 2.0',
     description='Awesome GAN toolkits based on PaddlePaddle',
+    long_description=readme(),
+    long_description_content_type='text/markdown',
     url='https://github.com/PaddlePaddle/PaddleGAN',
     download_url='https://github.com/PaddlePaddle/PaddleGAN.git',
     keywords=['gan paddlegan'],