from flask import Flask, request, jsonify
import joblib
app = Flask(__name__)
@app.route('/predict', methods=['POST'])
def post():
try:
if request.headers['useid'] == 'xxxx':
input_data = request.get_json()
model=joblib.load('model.pkl')
pre=model.predict(input_data)
output = {}
output['预测值'] = pre
return jsonify(output_dict)
else:
return jsonify(rstCode='False', rstMsg='请求失败!', error_reason='请求头useid错误!')
except Exception as e:
return jsonify(rstCode='False', rstMsg='请求失败!', error_reason=str(e))
此时,如果useid不是'xxxx‘的话,进行测试时就会出现'HTTP_USEID'报错, 一种修改方法就是将post代码改成
@app.route('/predict', methods=['POST'])
def post():
try:
input_data = request.get_json()
model=joblib.load('model.pkl')
pre=model.predict(input_data)
output = {}
output['预测值'] = pre
return jsonify(output_dict)
except Exception as e:
return jsonify(rstCode='False', rstMsg='请求失败!', error_reason=str(e))
去掉 if request.headers['useid'] == 'xxxx': 即可,或者将 'xxxx' 改成启服务人的'useid’。