Vantage training

serve.py

import joblib
from flask import Flask, request
import numpy as np
import os


app = Flask(__name__)
model = None


def load_model(path):
    return joblib.load(path)

@app.route("/api/v1/predict", methods=["POST"])
def predict():
    json = request.json
    input_image = np.array([json['image']])
    return {"result": int(model.predict(input_image)[0])}

if __name__ == "__main__":
    model = load_model(os.path.join(os.environ["MODEL_PATH"], "saved_model.pkl"))
    app.run(debug=True, host='0.0.0.0')