Serving the model in the Docker container