Creating a very basic deep-learning model in Docker