补充缺页判断

2024-10-10 11:24:16 +08:00
parent 15fe5d4f0d
commit e4b58e30c0
4 changed files with 56 additions and 8 deletions
--- a/services/paddle_services/init.py
+++ b/services/paddle_services/init.py
@@ -16,5 +16,6 @@ IE_KEY = {
    'admission_id': '住院号',
    'settlement_id': '医保结算单号码',
    'age': '年龄',
-    'upper_case_medical_expenses': '大写总额'
+    'upper_case_medical_expenses': '大写总额',
+    'page': '页码',
 }
--- a/services/paddle_services/ie_cost.py
+++ b/services/paddle_services/ie_cost.py
@@ -10,7 +10,7 @@ from utils import process_request

 app = Flask(__name__)
 COST_LIST_SCHEMA = tuple(IE_KEY[key] for key in [
-    'name', 'admission_date', 'discharge_date', 'medical_expenses'
+    'name', 'admission_date', 'discharge_date', 'medical_expenses', 'page'
 ])
 COST = Taskflow('information_extraction', schema=COST_LIST_SCHEMA, model='uie-x-base',
                task_path='model/cost_list_model', layout_analysis=False, precision='fp16')