in Uncategorized

How to make some prediction API using Keras + Flask in 50 lines of code

As everybody knows Pickle is the most used project to serialize objects in Python when we talk about Scikit-learn Machine Learning.
However, as most of you know, Keras do not have support for Pickle (and the official team does not recommend to use pickle or cPickle to save a Keras model).
There’s a bunch of workarounds to deal with this problem, like this great library called Keras Pickle Wrapper and this great post of Zach Moshe.
But, if you want to use the default options to save (serialize) your models and make some servicing of your models using a simple Flask API, you can use the following code (or get the code in github):
[code language=”python”]
import os
import keras
import numpy as np
from flask import jsonify
from flask import request
from flask import Flask
from keras.models import model_from_json
from keras.models import load_model
# Let’s startup the Flask application
app = Flask(__name__)
# Model reload from jSON:
print ‘Load model…’
json_file = open(‘models/keras_v1.00_model.json’, ‘r’)
loaded_model_json =
keras_model_loaded = model_from_json(loaded_model_json)
print ‘Model loaded…’
# Weights reloaded from .h5 inside the model
print ‘Load weights…’
print’Weights loaded…’
# Processing data from request and transform inside numpy array
def get_predict_data(data_unprocessed):
print’Load data…’
x = np.array(data_unprocessed)
print’Data loaded…’
return x
# URL that we’ll use to make predictions using get and post
def predict():
x = request.get_data()
x = x.split(",")
x = [x]
data_processed = get_predict_data(x)
y_hat = keras_model_loaded.predict(data_processed, batch_size=1, verbose=1)
return jsonify({‘prediction’: str(y_hat)}) # This is the result that will be returned in Flask
if __name__ == "__main__":
    # Choose the port
    port = int(os.environ.get(‘PORT’, 5000))
    # Run locally’′, port=port)
# CURL example for some prediction
# $ curl -X POST -d "6,0.0,1,3,2,1,0,0,0,12,0,1" "localhost:5000/predict"
References: Post totally inspired in Shuai, Loads Pickle, and I’m living contradiction posts.

Write a Comment


This site uses Akismet to reduce spam. Learn how your comment data is processed.