Update Dockerfile

Dockerfile

@@ -34,11 +34,16 @@ ENV HF_HOME=/models/huggingface \
 RUN mkdir -p /models/huggingface && chmod -R 777 /models/huggingface
 
 # --- Preload only N-ATLaS model ---
-RUN python -c "from huggingface_hub import snapshot_download; snapshot_download(repo_id='NCAIR1/N-ATLaS')"
+# HF_TOKEN is passed securely at runtime (not baked into image)
+ARG HF_TOKEN
+ENV HF_TOKEN=$HF_TOKEN
+
+RUN python -c "from huggingface_hub import snapshot_download; snapshot_download(repo_id='NCAIR1/N-ATLaS', use_auth_token='$HF_TOKEN')" \
+ && find /models/huggingface -name '*.lock' -delete
 
 # Copy project files
 COPY . .
 
 EXPOSE 7860
 
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "1"]
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "1"]