Update Dockerfile.inf2

Author: philschmid

dockerfiles/pytorch/Dockerfile.inf2

@@ -1,5 +1,5 @@
 # Build based on https://github.com/aws/deep-learning-containers/blob/master/huggingface/pytorch/inference/docker/2.1/py3/sdk2.18.0/Dockerfile.neuronx
-FROM ubuntu:20.04
+FROM ubuntu:20.04 as base
 
 LABEL maintainer="Hugging Face"
 
@@ -119,4 +119,4 @@ COPY src/huggingface_inference_toolkit/webservice_starlette.py webservice_starle
 # copy entrypoint and change permissions
 COPY --chmod=0755  scripts/entrypoint.sh entrypoint.sh
 
-ENTRYPOINT ["bash", "-c", "./entrypoint.sh"]
+ENTRYPOINT ["bash", "-c", "./entrypoint.sh"]