diff --git a/test.py b/test.py new file mode 100644 index 0000000..9ede1e1 --- /dev/null +++ b/test.py @@ -0,0 +1,17 @@ +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)