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})