feat: added docker build arguments to modal server

Author: limcheekin

Dockerfile-Modal

@@ -1,7 +1,8 @@
 FROM debian:bullseye-slim AS build-image
 
-#ARG MODEL
-#ENV MODEL=${MODEL}
+ARG MODEL
+ARG HOME
+ENV MODEL=${MODEL}
 COPY ${HOME}/download.sh ./
 
 # Install build dependencies
@@ -15,7 +16,7 @@ RUN chmod +x *.sh && \
 # Grab a fresh copy of the Python image
 FROM python:3.11-slim
 
-#ARG MODEL
+ARG MODEL
 
 RUN mkdir -p ${MODEL}
 COPY --from=build-image ${MODEL} ${MODEL}

open/text/embeddings/server/modal.py

@@ -5,12 +5,16 @@
 app = App(os.environ["APP_NAME"])
 
 image = Image.from_dockerfile(
-    "Dockerfile-Modal", force_build=True
+    "Dockerfile-Modal", 
+    force_build=True,
+    build_args={
+        "MODEL": os.environ["MODEL"],
+        "HOME": os.environ["HOME"],
+    }
 ).env({
     "MODEL": os.environ["MODEL"],
     "NORMALIZE_EMBEDDINGS": os.environ["NORMALIZE_EMBEDDINGS"],
     "VERBOSE": os.environ["VERBOSE"],
-    "HOME": os.environ["HOME"],
     "HF_HOME": "/tmp/hf_home",
 })