Update Dockerfile

Dockerfile

@@ -1,9 +1,11 @@
 FROM python:3.10-slim
 
+# Install system dependencies
 RUN apt-get update && apt-get install -y git
 
 WORKDIR /app
 
+# Copy requirements and install
 COPY requirements.txt .
 RUN pip install --upgrade pip && pip install -r requirements.txt
 
@@ -12,16 +14,19 @@ RUN git clone https://github.com/earthspecies/NatureLM-audio.git
 RUN sed -i '/beans-zero/d' NatureLM-audio/pyproject.toml
 RUN pip install -e NatureLM-audio
 
-# Set cache envs to writable location
+# Set HuggingFace and Torch cache to a writable directory
 ENV HF_HOME=/app/.cache/huggingface
 ENV TRANSFORMERS_CACHE=/app/.cache/transformers
 ENV TORCH_HOME=/app/.cache/torch
 ENV XDG_CACHE_HOME=/app/.cache
+
+# Set HuggingFace token for gated model access (injected from Space secret)
+ENV HF_TOKEN=${HF_TOKEN}
 RUN mkdir -p /app/.cache && chmod -R 777 /app/.cache
 
+# Copy your FastAPI app
 COPY main.py .
 
 EXPOSE 7860
 
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]