diff --git a/main.py b/main.py
index edaf0d6..6ac048f 100644
--- a/main.py
+++ b/main.py
@@ -29,14 +29,14 @@
 #             paddle.device.cuda.empty_cache()
 import logging.config
 
+from config import photo_review as pr
 from config.log import LOGGING_CONFIG
-from config.photo_review import RETRY_TIME
 from photo_review import photo_review
 
 if __name__ == '__main__':
     logging.config.dictConfig(LOGGING_CONFIG)
     log = logging.getLogger()
 
-    for _ in range(RETRY_TIME + 1):
+    for _ in range(pr.RETRY_TIME + 1):
         log.info("【照片审核关键信息抽取】开始")
         photo_review.main()