增加对结果长度的判断,减少数据库添加失败的情况
This commit is contained in:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user