diff --git a/main.py b/main.py index fa2bcc3..d46c617 100644 --- a/main.py +++ b/main.py @@ -2,15 +2,15 @@ import logging.config import os 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, SEND_ERROR_EMAIL, CUDA_VISIBLE_DEVICES -from photo_review.photo_review import main os.environ["CUDA_VISIBLE_DEVICES"] = CUDA_VISIBLE_DEVICES +import paddle +from auto_email.error_email import send_an_error_email +from config.log import LOGGING_CONFIG +from photo_review.photo_review import main + # 项目必须从此处启动,否则代码中的相对路径可能导致错误的发生 if __name__ == '__main__': logging.config.dictConfig(LOGGING_CONFIG) diff --git a/photo_review/__init__.py b/photo_review/__init__.py index 44a965c..01fea81 100644 --- a/photo_review/__init__.py +++ b/photo_review/__init__.py @@ -1,12 +1,8 @@ -import os - from paddlenlp import Taskflow from config.keys import PATIENT_NAME, ADMISSION_DATE, DISCHARGE_DATE, MEDICAL_EXPENSES, PERSONAL_CASH_PAYMENT, \ PERSONAL_ACCOUNT_PAYMENT, PERSONAL_FUNDED_AMOUNT, MEDICAL_INSURANCE_TYPE, HOSPITAL, DEPARTMENT, DOCTOR -from config.photo_review import IE_BATCH_SIZE, LAYOUT_ANALYSIS, CUDA_VISIBLE_DEVICES - -os.environ["CUDA_VISIBLE_DEVICES"] = CUDA_VISIBLE_DEVICES +from config.photo_review import IE_BATCH_SIZE, LAYOUT_ANALYSIS # 基本医保结算单 settlement_list_schema = \