From 5498d625422241759dd344996812fdef4738f879 Mon Sep 17 00:00:00 2001 From: liuyebo <1515783401@qq.com> Date: Wed, 5 Jun 2024 16:57:04 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E7=89=88=E9=9D=A2=E5=88=86?= =?UTF-8?q?=E6=9E=90?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- photo_review/photo_review.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/photo_review/photo_review.py b/photo_review/photo_review.py index 1fc9998..890ee25 100644 --- a/photo_review/photo_review.py +++ b/photo_review/photo_review.py @@ -28,7 +28,8 @@ def information_extraction(schema, phrecs, task_path): for phrec in phrecs: pic_path = get_private_url(phrec.cfjaddress) if pic_path: - ie = Taskflow("information_extraction", schema=schema, model="uie-x-base", task_path=task_path) + ie = Taskflow("information_extraction", schema=schema, model="uie-x-base", task_path=task_path, + layout_analysis=True) # 批量抽取写法:(ie([{"doc": "./data/6.jpg"}, {"doc": "./data/7.jpg"}]) result = ie({"doc": pic_path})