Flask部署Keras模型

server端

# Flask
import flask
import numpy as np
from flask import Flask

from keras.models import load_model

app = Flask(__name__)

model.load_model("yourmodel.h5")

# 模型预测，返回预测结果
def model_predict(text, model):
    res = model.predict(text)
    return res

# 真神奇！！！不加下面两句就报错，加了就正常调用
predict_text = "CONTAINER"
s = model_predict(predict_text, model)

@app.route("/predict", methods=["GET", "POST"])
def predict():
    data = {"success": False}
    params = flask.request.json
    if params is None:
        params = flask.request.args

    # 若发现参数，则返回预测值
    if params:
        text = params['text']
        preds = model_predict(text, model)
        data["prediction"] = preds
        data["success"] = True

    # 返回Json格式的响应
    return flask.jsonify(data)
    
if __name__ == '__main__':
    app.run(debug=False, host='0.0.0.0', port=5000)

client端

import requests

data = {
    "text": "CONTAINER PLASTIC SCRAP"
}
header = {"Content-Type": "application/json; charset=utf-8"}
url = "http://127.0.0.1:5000/predict?"
response = requests.post(url, json=data, headers=header)
res = response.json()
print(res)
print("Prediction:", res['prediction'])
print("Success:", res['success'])

合作联系电话（VX）18012411994