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)