Skip to content

Instantly share code, notes, and snippets.

@hzitoun
Last active May 18, 2020 17:30
Show Gist options
  • Save hzitoun/032a9cced5981e2f607f0d28537037a1 to your computer and use it in GitHub Desktop.
Save hzitoun/032a9cced5981e2f607f0d28537037a1 to your computer and use it in GitHub Desktop.
Code for my medium story Serve a Deep Learning Image Classification Model Written in TensorFlow 2 as a REST API and Dockerize it! (with GPU support)
name: tf2-image-classifier-api
channels:
- anaconda
- conda-forge
- defaults
dependencies:
- pip==20.1
- numpy==1.18.4
- flask==1.1.2
- matplotlib==3.2.1
- seaborn==0.10.1
- scikit-learn==0.23
- notebook==6.0.3
- ipywidgets==7.5.1
- waitress==1.4.3
- pip:
- tensorflow-gpu==2.2.0
- opencv-python==4.2.0.34
- tqdm==4.46.0
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment