Fixed export of MODEL_PATH

app/utils/download_model.py

@@ -2,6 +2,9 @@ import os
 import requests
 from transformers import AutoModel
 
+'''
+DO NOT PRINT ANYTHING OTHER THAN MODEL PATH IN THIS SCRIPT, THIS IS BEING EXPORTED BY DOCKER ENTRY POINT!
+'''
 def model_file_exists_and_valid(model_file_path):
     # Check if the model file exists and has a size greater than 0
     return os.path.exists(model_file_path) and os.path.getsize(model_file_path) > 0
@@ -17,7 +20,7 @@ def download_hf_model():
     
     if model_file_exists_and_valid(model_file_path):
         print(f"Model {model_name} already downloaded.")
-        os.environ['MODEL_PATH'] = model_file_path
+        print(model_file_path)
         return
     
     # Authenticate with Hugging Face using the token, if available
@@ -30,7 +33,7 @@ def download_hf_model():
     model = AutoModel.from_pretrained(model_name)
     model.save_pretrained(model_dir)
     print(f"Model {model_name} downloaded and saved to {model_dir}")
-    os.environ['MODEL_PATH'] = model_file_path
+    print(model_file_path)
 
 def download_gguf_model():
     model_name = os.getenv("HF_MODEL_NAME")
@@ -42,7 +45,7 @@ def download_gguf_model():
     
     if model_file_exists_and_valid(model_file_path):
         print(f"Model {model_name} already downloaded.")
-        os.environ['MODEL_PATH'] = model_file_path
+        print(model_file_path)
         return
     
     print(f"Downloading model from {model_url}...")
@@ -53,7 +56,7 @@ def download_gguf_model():
         print(f"Model downloaded and saved to {model_file_path}")
     else:
         print(f"Failed to download the model. Status code: {response.status_code}")
-    os.environ['MODEL_PATH'] = model_file_path
+    print(model_file_path)
 
 def download_model():
     model_class = os.getenv("MODEL_CLASS")

entrypoint.sh

@@ -18,7 +18,9 @@ if [ ! -d "$MODEL_DIR" ]; then
 
     # Execute the download_model script
     echo "Downloading the model..."
-    python download_model.py || { echo "Model download failed"; exit 1; }
+    MODEL_PATH=$(python download_model.py) || { echo "Model download failed"; exit 1; }
+    export MODEL_PATH
+
 
     # Navigate back to the app directory
     cd /home/user/app