From 80fef129a96262dc3054b219fd94c37d56c0ecc7 Mon Sep 17 00:00:00 2001 From: zty Date: Tue, 29 Oct 2024 17:03:38 +0800 Subject: [PATCH] =?UTF-8?q?fix=20:=20=E4=BF=AE=E6=94=B9=20main?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- main.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/main.py b/main.py index e0d805c..0f4d14a 100644 --- a/main.py +++ b/main.py @@ -284,7 +284,7 @@ async def extract_info( ocr_text = "\n".join( [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 file_type == "patent": @@ -321,10 +321,11 @@ async def extract_info( elif file_type == "social": prompt = f"我有以下文本,是一个社会标准的内容。请按项目名称，获奖单位，奖励等级，获奖人，奖励年度,证书编号,颁发日期为key的json格式返回数据,注意只返回json数据。文本如下：{ocr_text}" elif file_type=="building": - prompt = f"我有以下文本,是一个建筑标准的内容。请按项目名称，奖励等级，获奖单位，证书编号，颁发日期为key的json格式返回数据,注意只返回json数据。文本如下：{ocr_text}" + prompt = f"我有以下文本,是一个建筑标准的内容。请按项目名称，奖励等级，获奖单位，证书编号，颁发日期为key的json格式返回数据,注意只返回json数据。文本如下：{ocr_text}" + else: 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( conf.CHAT_API, json={"model": conf.CHAT_MODEL, "prompt": prompt, "stream": False},