Echo/fcb_photo_review
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

调整文本分类的判空

Browse Source
This commit is contained in:
liuyebo
2024-10-18 10:19:33 +08:00
parent 9556da47e9
commit 61a7802674
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
services/paddle_services/clas_text.py
View File

@@ -17,9 +17,9 @@ CLAS = Taskflow('zero_shot_text_classification', model='utc-xbase', schema=schem
def main(): def main():
text = request.form.get('text') text = request.form.get('text')
cls_result = CLAS(text) cls_result = CLAS(text)
if not cls_result: cls_result = cls_result[0].get('predictions')
return None if cls_result:
cls_result = cls_result[0].get('predictions')[0] cls_result = cls_result[0]
if cls_result['score'] < 0.8: if cls_result['score'] < 0.8:
logging.info(f"识别结果置信度{cls_result['score']}过低！text: {text}") logging.info(f"识别结果置信度{cls_result['score']}过低！text: {text}")
return None return None