diff --git a/model_docker_images/inference/Dockerfile b/model_docker_images/inference/Dockerfile index 6433484bd..fab6d3f12 100644 --- a/model_docker_images/inference/Dockerfile +++ b/model_docker_images/inference/Dockerfile @@ -24,4 +24,7 @@ EXPOSE 8080 ENV PYTHONUNBUFFERED=TRUE # SageMaker will look for this -CMD ["serve"] \ No newline at end of file +CMD ["serve"] + +# Required label for SageMaker pipeline models +LABEL com.amazonaws.sagemaker.capabilities.accept-bind-to-port=true \ No newline at end of file