Entrypoint to load model

.env

@@ -1 +1 @@
-EC_ADMIN_PWD='$2b$12$zybxm7XMoGCVV3ovNDcXt.r2QJUhtj7miYfEfuBw9UGqViTIRFg72'
+EC_ADMIN_PWD='$2b$12$zybxm7XMoGCVV3ovNDcXt.r2QJUhtj7miYfEfuBw9UGqViTIRFg72'

app/utils/download_model.py

@@ -0,0 +1,24 @@
+# /home/user/app/utils/download_model.py
+import os
+from transformers import AutoModel
+
+def download_model():
+    # Use environment variables for the model name and directory
+    model_name = os.getenv("HF_MODEL_NAME")
+    model_dir = f"/home/user/data/models/{model_name}"
+
+    # Authenticate with Hugging Face using the token, if available
+    hf_token = os.getenv("HF_TOKEN")
+    if hf_token:
+        from huggingface_hub import HfFolder
+        HfFolder.save_token(hf_token)  # Save the token for later use by the library
+
+    # Download the model
+    print(f"Downloading model: {model_name}...")
+    model = AutoModel.from_pretrained(model_name)
+    model.save_pretrained(model_dir)
+    print(f"Model {model_name} downloaded and saved to {model_dir}")
+
+if __name__ == "__main__":
+    download_model()
+

entrypoint.sh

@@ -0,0 +1,26 @@
+#!/bin/bash
+
+# Authenticate with Hugging Face
+export HF_HOME=/home/user/data/hf_cache
+
+echo "Using Hugging Face API token for authentication"
+export HF_TOKEN=${HF_TOKEN}
+
+# Use the environment variable for the model name
+MODEL_DIR="/home/user/data/models/${HF_MODEL_NAME}"
+MODEL_URL=${HF_MODEL_URL}  # Ensure consistent variable naming
+
+# Download the model if it does not exist
+if [ ! -d "$MODEL_DIR" ]; then
+    echo "Model not found. Downloading ${HF_MODEL_NAME} from ${MODEL_URL}..."
+    mkdir -p "$MODEL_DIR"  # Ensure the directory exists before downloading
+    wget "$MODEL_URL" -P "$MODEL_DIR" || {
+        echo "Failed to download model from ${MODEL_URL}"
+        exit 1  # Exit if download fails
+    }
+else 
+    echo "Model ${HF_MODEL_NAME} already present."
+fi
+
+# Execute the main command of the container
+exec "$@"