增加对结果长度的判断,减少数据库添加失败的情况

This commit is contained in:
2024-05-21 09:28:38 +08:00
parent 898059ba0d
commit 72b40f4387

View File

@@ -27,10 +27,13 @@ def information_extraction(schema, phrecs, task_path):
ie = Taskflow("information_extraction", schema=schema, model="uie-x-base", task_path=task_path)
result = ie({"doc": pic_path})
result_json = json.dumps(result, ensure_ascii=False)
if len(result_json) > 5000:
result_json = result_json[:5000]
# 提取完保存每张图片的结果
session = MysqlSession()
zx_ocr = ZxOcr(pk_phhd=phrec.pk_phhd, pk_phrec=phrec.pk_phrec, cfjaddress=phrec.cfjaddress,
content=json.dumps(result, ensure_ascii=False))
content=result_json)
session.add(zx_ocr)
session.commit()
session.close()