Building a Docker image for TF serving