Files
fcb_photo_review/test.py
2024-07-06 15:21:58 +08:00

18 lines
579 B
Python

from paddlenlp import Taskflow
SCHEMA = ["患者姓名", "入院日期", "出院日期", "费用总额", "个人现金支付", "个人账户支付", "自费金额", "医保类型",
"住院号", "医保结算单号码"]
IE = Taskflow("information_extraction", schema=SCHEMA, model="uie-x-base",
task_path="config/model/settlement_list_model", layout_analysis=True)
def main():
i = 0
try:
while 1:
IE({"doc": "visual_model_test/img/PH20240511000638_1_094306_1.jpg"})
i += 1
except Exception:
print(i)