From 11e64953045fa615bfdb786bfe6f8975603ce8c3 Mon Sep 17 00:00:00 2001
From: Justin Merrell <merrelljustin@gmail.com>
Date: Fri, 8 Dec 2023 20:28:44 -0500
Subject: [PATCH] Update Dockerfile

---
 Dockerfile | 1 +
 1 file changed, 1 insertion(+)

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 \