Basic Flask skeleton for SageMaker Docker file

app.py

from flask import Flask, request, abort

@app.route('/ping', methods=['GET', 'POST'])
def ping():
    return 'ping is pong'

@app.route('/invocations', methods=['POST'])
def invocation():
    data = request.get_json()
    if not data:
        abort(418, "Send some data plz")
    norms = data.get('norms', [])

    icons = .... (the prediction code)
    
    return json.dumps({'icons': icons})

if __name__ == '__main__':
    app.run(debug=True, host='0.0.0.0', port=8080)