...

Dockerfile

@@ -1,36 +1,23 @@
+# Dockerfile
 FROM python:3.10-slim
 WORKDIR /app
 
-# 1) Install HF tooling & your deps
+# 1) Install HF tooling & runtime deps
 RUN pip install --no-cache-dir huggingface_hub
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# 2) Create a cache dir
-RUN mkdir -p /app/model_cache
-
-# 3) Download the model into model_cache (build-time)
+# 2) Download model at build-time directly into model_cache
 RUN python - <<EOF
 from huggingface_hub import snapshot_download
-# downloads into /app/models--numind--NuExtract-1.5-tiny/snapshots/<hash>/
-tmp = snapshot_download(
+# This writes ALL repo files under /app/model_cache
+snapshot_download(
     repo_id="numind/NuExtract-1.5-tiny",
-    cache_dir="/app",
-    local_dir="model_cache",
-    local_dir_use_symlinks=False
+    local_dir="/app/model_cache"
 )
-# flatten: move from the hash-folder up into /app/model_cache
-import os, shutil, glob
-root = os.path.join("/app/model_cache","models--numind--NuExtract-1.5-tiny","snapshots")
-# there should be exactly one subdir under root
-sub = glob.glob(os.path.join(root, "*"))[0]
-for fname in os.listdir(sub):
-    shutil.move(os.path.join(sub, fname), "/app/model_cache")
-# clean up
-shutil.rmtree(os.path.join("/app/model_cache","models--numind--NuExtract-1.5-tiny"))
 EOF
 
-# 4) Copy your code
+# 3) Copy your FastAPI code
 COPY . .
 
 ENV PORT=7860

app.py

@@ -23,11 +23,11 @@ model_name = "numind/NuExtract-1.5-tiny"
 # You can also symlink or copy it to /app/model directly in Dockerfile.
 
 
-MODEL_PATH = "/app/model_cache"
-# model_cache_path = snapshot_download(
-#     repo_id="numind/NuExtract-1.5-tiny",
-#     cache_dir=MODEL_PATH
-# )
+# MODEL_PATH = "/app/model_cache"
+model_cache_path = snapshot_download(
+    repo_id="numind/NuExtract-1.5-tiny",
+    local_dir="/app/model_cache"            # <-- direct destination
+)
 
 print(">>> MODEL CACHE PATH:", MODEL_PATH, os.listdir(MODEL_PATH))