调整文本分类的精度,fp16也许只能接收1024个字符
This commit is contained in:
@@ -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'])
|
||||
|
||||
Reference in New Issue
Block a user