文本分类增加判空

services/paddle_services/clas_text.py

@@ -9,7 +9,7 @@ from utils import process_request
 app = Flask(__name__)
 schema = ['基本医保结算单', '出院记录', '费用清单']
 CLAS = Taskflow('zero_shot_text_classification', model='utc-xbase', schema=schema,
-                task_path='model/text_classification', precision='fp32')
+                task_path='model/text_classification', precision='fp16')
 
 
 @app.route('/', methods=['POST'])
@@ -17,6 +17,8 @@ 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]
     if cls_result['score'] < 0.8:
         logging.info(f"识别结果置信度{cls_result['score']}过低！text: {text}")