hzitoun/save_model.py

Created May 18, 2020 17:03

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/hzitoun/4a98d8f16d3c4fc2b0e5188e189e72cb.js"></script>
Save hzitoun/4a98d8f16d3c4fc2b0e5188e189e72cb to your computer and use it in GitHub Desktop.

Download ZIP

Source code for medium story 🚀Serve a TensorFlow 2 Deep Learning Image Classifier as a REST API and Dockerize it! (with GPU support)

Raw

save_model.py

model.save('../models/saved_image_classifier/')

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment