From 61a7802674156305734c99a98159a5559552482e Mon Sep 17 00:00:00 2001 From: liuyebo <1515783401@qq.com> Date: Fri, 18 Oct 2024 10:19:33 +0800 Subject: [PATCH] =?UTF-8?q?=E8=B0=83=E6=95=B4=E6=96=87=E6=9C=AC=E5=88=86?= =?UTF-8?q?=E7=B1=BB=E7=9A=84=E5=88=A4=E7=A9=BA?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- services/paddle_services/clas_text.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/services/paddle_services/clas_text.py b/services/paddle_services/clas_text.py index 67fd5a4..9eb77fd 100644 --- a/services/paddle_services/clas_text.py +++ b/services/paddle_services/clas_text.py @@ -17,9 +17,9 @@ CLAS = Taskflow('zero_shot_text_classification', model='utc-xbase', schema=schem def main(): text = request.form.get('text') cls_result = CLAS(text) - if not cls_result: - return None - cls_result = cls_result[0].get('predictions')[0] + cls_result = cls_result[0].get('predictions') + if cls_result: + cls_result = cls_result[0] if cls_result['score'] < 0.8: logging.info(f"识别结果置信度{cls_result['score']}过低!text: {text}") return None