Fixing model name not being passed into build

Author: jianshen92

src/clip_api_service/build.py

@@ -1,5 +1,7 @@
 from __future__ import annotations
 
+from clip_api_service.models import MODEL_ENV_VAR_KEY
+
 import os
 
 
@@ -13,6 +15,11 @@ def build_bento(model_name: str | None = None, use_gpu: bool = False):
 
     bento_file = clip_module.bentofile_path(use_gpu=use_gpu)
 
+    # Setting Model name as environment variable to pass 
+    # into subprocess for bento build
+    if model_name:
+        os.environ[MODEL_ENV_VAR_KEY] = model_name
+
     # Get parent directory path
     build_ctx = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
     bentoml.bentos.build_bentofile(bento_file, build_ctx=build_ctx)