补充缺页判断

This commit is contained in:
2024-10-10 11:24:16 +08:00
parent 15fe5d4f0d
commit e4b58e30c0
4 changed files with 56 additions and 8 deletions

View File

@@ -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')