update docker files for spaces deployment

Dockerfile

@@ -2,11 +2,13 @@ FROM python:3.10-slim
 
 # Add build argument
 ARG HUGGINGFACE_API_KEY
+ENV HUGGINGFACE_API_KEY=${HUGGINGFACE_API_KEY}
 
 # System dependencies
 RUN apt-get update && apt-get install -y \
     build-essential \
     curl \
+    git \
     && rm -rf /var/lib/apt/lists/*
 
 # Set working directory
@@ -19,9 +21,18 @@ RUN pip install --no-cache-dir -r requirements.txt
 # Copy project files
 COPY . .
 
-# Environment variables
-ENV PYTHONUNBUFFERED=1
-ENV HUGGINGFACE_API_KEY=${HUGGINGFACE_API_KEY}
+# # Environment variables
+# ENV PYTHONUNBUFFERED=1
+# ENV HUGGINGFACE_API_KEY=${HUGGINGFACE_API_KEY}
+
+# HuggingFace login at container startup
+# RUN --mount=type=secret,id=hf_token \
+#     HUGGINGFACE_API_KEY=$(cat /run/secrets/hf_token) && \
+#     huggingface-cli login --token $HUGGINGFACE_API_KEY
+
+RUN mkdir -p ~/.huggingface && \
+    echo "${HUGGINGFACE_API_KEY}" > ~/.huggingface/token && \
+    huggingface-cli login --token ${HUGGINGFACE_API_KEY}
 
 # Expose port
 EXPOSE 7860

docker-compose.yml

@@ -8,7 +8,10 @@ services:
     ports:
       - "7860:7860"
     environment:
+      # - PYTHONUNBUFFERED=1
       - HUGGINGFACE_API_KEY=${HUGGINGFACE_API_KEY}
+    secrets:
+      - hf_token
     deploy:
       resources:
         reservations: