zcdsj
/
cert_recog
5
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix : 修改 main

This commit is contained in:
zty 2024-10-29 17:03:38 +08:00
parent bac3ae7fe3
commit 80fef129a9
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
main.py
View File

@ -284,7 +284,7 @@ async def extract_info(
ocr_text = "\n".join( ocr_text = "\n".join(
[line[1][0] for res in result if res is not None for line in res] [line[1][0] for res in result if res is not None for line in res]
) )
# print("ocr_text", ocr_text) print("ocr_text", ocr_text)
# 提取信息 # 提取信息
if extract_method == "re": if extract_method == "re":
if file_type == "patent": if file_type == "patent":
@ -322,9 +322,10 @@ async def extract_info(
prompt = f"我有以下文本,是一个社会标准的内容。请按项目名称，获奖单位，奖励等级，获奖人，奖励年度,证书编号,颁发日期为key的json格式返回数据,注意只返回json数据。文本如下：{ocr_text}" prompt = f"我有以下文本,是一个社会标准的内容。请按项目名称，获奖单位，奖励等级，获奖人，奖励年度,证书编号,颁发日期为key的json格式返回数据,注意只返回json数据。文本如下：{ocr_text}"
elif file_type=="building": elif file_type=="building":
prompt = f"我有以下文本,是一个建筑标准的内容。请按项目名称，奖励等级，获奖单位，证书编号，颁发日期为key的json格式返回数据,注意只返回json数据。文本如下：{ocr_text}" prompt = f"我有以下文本,是一个建筑标准的内容。请按项目名称，奖励等级，获奖单位，证书编号，颁发日期为key的json格式返回数据,注意只返回json数据。文本如下：{ocr_text}"
else:
raise HTTPException( raise HTTPException(
400, detail="Invalid file type. Please choose 'standard' or 'patent'." 400, detail="Invalid file type. Please choose 'standard' or 'patent'."
) )
r = requests.post( r = requests.post(
conf.CHAT_API, conf.CHAT_API,
json={"model": conf.CHAT_MODEL, "prompt": prompt, "stream": False}, json={"model": conf.CHAT_MODEL, "prompt": prompt, "stream": False},