Update Dockerfile

Dockerfile

@@ -2,31 +2,28 @@ FROM python:3.10-slim
 
 WORKDIR /app
 
-# Install git and git-lfs to handle large files
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
     git \
     git-lfs \
     && git lfs install \
     && rm -rf /var/lib/apt/lists/*
 
-# Install requirements
+# Install python requirements
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# ARG allows passing the token during build time
-ARG HF_TOKEN
-ENV HF_TOKEN=${HF_TOKEN}
-
-# Use the CLI to download only the model files into the /app/model folder
-# This skips the git history, README, and attributes automatically
-RUN huggingface-cli download CooLLaMACEO/Overflow-100B \
+# --- THE BUILD-TIME SECRET MOUNT ---
+# This line accesses your HF_TOKEN secret to download the model
+RUN --mount=type=secret,id=HF_TOKEN,mode=0444,required=true \
+    huggingface-cli download CooLLaMACEO/Overflow-100B \
+    --token $(cat /run/secrets/HF_TOKEN) \
     --local-dir /app/model \
     --local-dir-use-symlinks False \
-    --exclude "README.md" ".gitattributes" \
-    --token ${HF_TOKEN}
+    --exclude "README.md" ".gitattributes"
 
-# Copy your local app files
-COPY app.py modeling_overflow.py configuration_overflow.py ./
+# Copy the rest of your app code
+COPY . .
 
 # Force CPU mode
 ENV CUDA_VISIBLE_DEVICES=""