调整文本分类的精度,fp16也许只能接收1024个字符

This commit is contained in:
2024-10-18 08:56:12 +08:00
parent 74920869e7
commit 8c47beb00c

View File

@@ -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='fp16')
task_path='model/text_classification', precision='fp32')
@app.route('/', methods=['POST'])