From f9ef9f50510544e68c111dd104a079d764cc6503 Mon Sep 17 00:00:00 2001
From: liuyebo <1515783401@qq.com>
Date: Fri, 21 Jun 2024 15:44:24 +0800
Subject: [PATCH] =?UTF-8?q?=E5=86=B2=E7=AA=81=E6=8E=92=E6=9F=A5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 main.py | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/main.py b/main.py
index edaf0d6..2efff18 100644
--- a/main.py
+++ b/main.py
@@ -28,15 +28,25 @@
 #             # 释放显存
 #             paddle.device.cuda.empty_cache()
 import logging.config
+import traceback
 
+import paddle
+
+from auto_email.error_email import send_an_error_email
 from config.log import LOGGING_CONFIG
-from config.photo_review import RETRY_TIME
+from config.photo_review import SEND_ERROR_EMAIL
 from photo_review import photo_review
 
 if __name__ == '__main__':
     logging.config.dictConfig(LOGGING_CONFIG)
     log = logging.getLogger()
 
-    for _ in range(RETRY_TIME + 1):
+    try:
         log.info("【照片审核关键信息抽取】开始")
         photo_review.main()
+    except Exception as e:
+        log.error(traceback.format_exc())
+        if SEND_ERROR_EMAIL:
+            send_an_error_email(program_name='照片审核关键信息抽取脚本', error_name=repr(e), error_detail=traceback.format_exc())
+        # 释放显存
+        paddle.device.cuda.empty_cache()