diff --git a/Dockerfile b/Dockerfile
index 0bde5c3..091b4bc 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -29,6 +29,7 @@ ADD src .
 ARG MODEL_NAME=""
 ARG MODEL_BASE_PATH=""
 ARG HUGGING_FACE_HUB_TOKEN=""
+ENV HUGGING_FACE_HUB_TOKEN=$HUGGING_FACE_HUB_TOKEN
 
 # Conditionally run download_model.py
 RUN if [ -n "$MODEL_NAME" ] && [ -n "$MODEL_BASE_PATH" ]; then \