Update download_model.py

download_model.py

@@ -1,8 +1,18 @@
-from huggingface_hub import hf_hub_download
 import os
+from huggingface_hub import hf_hub_download
+
+# Force writable cache path
+os.environ["HF_HOME"] = "/tmp/hf_cache"
+os.environ["HF_HUB_CACHE"] = "/tmp/hf_cache"
 
-repo_id = os.environ.get("MODEL_REPO")
-filename = os.environ.get("MODEL_FILE")
+repo_id = "TheBloke/phi-2-GGUF"
+filename = "phi-2.Q4_K_M.gguf"
 
-print("Downloading model:", repo_id, filename)
-hf_hub_download(repo_id=repo_id, filename=filename, local_dir=".", local_dir_use_symlinks=False)
+# Download to current working dir
+hf_hub_download(
+    repo_id=repo_id,
+    filename=filename,
+    cache_dir="/tmp/hf_cache",  # override .cache
+    local_dir=".",
+    local_dir_use_symlinks=False
+)