From 2005b038b9375290589cd7be0d6ac3fef25aa0c6 Mon Sep 17 00:00:00 2001 From: liuyebo <1515783401@qq.com> Date: Fri, 21 Jun 2024 15:28:08 +0800 Subject: [PATCH] =?UTF-8?q?=E8=B0=83=E6=95=B4=E4=BF=A1=E6=81=AF=E6=8A=BD?= =?UTF-8?q?=E5=8F=96=E6=96=B9=E6=B3=95=E7=9A=84=E7=94=9F=E6=88=90=E4=BD=8D?= =?UTF-8?q?=E7=BD=AE?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- config/photo_review.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/config/photo_review.py b/config/photo_review.py index b85d4d0..523012b 100644 --- a/config/photo_review.py +++ b/config/photo_review.py @@ -22,3 +22,17 @@ LAYOUT_ANALYSIS = False # 可使用的GPU CUDA_VISIBLE_DEVICES = "1" + +# 基本医保结算单 +SETTLEMENT_IE = Taskflow("information_extraction", schema=SETTLEMENT_LIST_SCHEMA, model="uie-x-base", + task_path="config/model/settlement_list_model", layout_analysis=LAYOUT_ANALYSIS, + batch_size=IE_BATCH_SIZE) + +# 出院记录 +DISCHARGE_IE = Taskflow("information_extraction", schema=DISCHARGE_RECORD_SCHEMA, model="uie-x-base", + task_path="config/model/discharge_record_model", layout_analysis=LAYOUT_ANALYSIS, + batch_size=IE_BATCH_SIZE) + +# 费用清单 +COST_IE = Taskflow("information_extraction", schema=COST_LIST_SCHEMA, model="uie-x-base", + task_path="config/model/cost_list_model", layout_analysis=LAYOUT_ANALYSIS, batch_size=IE_BATCH_SIZE)