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

fix: 修改模型的语句

This commit is contained in:
zty 2024-10-29 16:39:59 +08:00
parent d97a4d436f
commit bac3ae7fe3
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
main.py
View File

@ -314,7 +314,14 @@ async def extract_info(
prompt = f"我有以下文本,是一个专利的内容。请按专利名称，发明人，专利号，专利申请日， 专利权人，授权公告号，授权公告日为key的json格式返回数据,注意只返回json数据。文本如下：{ocr_text}"
elif file_type == "standard":
prompt = f"我有以下文本,是一个标准的内容。请按标准号，中文名称，英文名称，发布部门，发布日期，实施日期，提出单位，归口单位，起草单位，起草人为key的json格式返回数据,注意只返回json数据。文本如下：{ocr_text}"
else:
elif file_type == "country":
prompt = f"我有以下文本,是一个国家标准的内容。请按项目名称，奖励等级，获奖者，证书号，颁发日期为key的json格式返回数据,注意只返回json数据。文本如下：{ocr_text}"
elif file_type == "province":
prompt = f"我有以下文本,是一个省级标准的内容。请按项目名称，奖励等级，获奖者，证书编号，颁发日期为key的json格式返回数据,注意只返回json数据。文本如下：{ocr_text}"
elif file_type == "social":
prompt = f"我有以下文本,是一个社会标准的内容。请按项目名称，获奖单位，奖励等级，获奖人，奖励年度,证书编号,颁发日期为key的json格式返回数据,注意只返回json数据。文本如下：{ocr_text}"
elif file_type=="building":
prompt = f"我有以下文本,是一个建筑标准的内容。请按项目名称，奖励等级，获奖单位，证书编号，颁发日期为key的json格式返回数据,注意只返回json数据。文本如下：{ocr_text}"
raise HTTPException(
400, detail="Invalid file type. Please choose 'standard' or 'patent'."
)