# MNIST Recognition Flask app by akashdeepjassal changeable model build ``` my repo: https://github.com/Qiaoster/mnist-flask.git ``` ``` original repo: https://github.com/akashdeepjassal/mnist-flask/blob/master/model/model.json ``` upgraded from tensorflow 1.4, can easily swap in your own model. Prediction set to (1, 28, 28, 1). ## How to swap model Step 1: Save your model has an .h5 file and save it in mnist-flask folder. ![](https://i.imgur.com/mXxAL8Y.png) (as you can see there're already two models here, my_model.h5 and cnn_model.h5) Step 2: Open load.py file under mnist-flask\model folder, and edit model_name variable to the file name of your own model. ![](https://i.imgur.com/uXHhPsL.png) Done!