添加版面分析

2024-06-05 16:57:04 +08:00
parent fbd6834dd5
commit 5498d62542
1 changed files with 2 additions and 1 deletions
--- 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})