Update Dockerfile

Dockerfile

@@ -10,7 +10,7 @@ ENV PYTHONUNBUFFERED=1 \
 # Set working directory
 WORKDIR /app
 
-# Copy required files
+# Copy the required files
 COPY requirements.txt .  
 COPY main.py .  
 
@@ -19,16 +19,19 @@ RUN python -m venv /app/venv && \
     /app/venv/bin/pip install --no-cache-dir --upgrade pip && \
     /app/venv/bin/pip install --no-cache-dir -r requirements.txt
 
-
 # Ensure the model cache directory exists
 RUN mkdir -p $HF_HOME
 
+# Add Hugging Face token as an environment variable in the container (this will be injected by Hugging Face Spaces' secrets management)
+ARG HF_TOKEN
+ENV HF_TOKEN=${HF_TOKEN}
+
 # Pre-download models (handle errors gracefully)
 RUN /app/venv/bin/python -c "from transformers import pipeline; \
-    pipeline('sentiment-analysis', model='tabularisai/multilingual-sentiment-analysis')" || echo "Failed to download model 1"
+    pipeline('sentiment-analysis', model='tabularisai/multilingual-sentiment-analysis', use_auth_token='$HF_TOKEN')" || echo 'Failed to download model 1'
 
 RUN /app/venv/bin/python -c "from transformers import pipeline; \
-    pipeline('sentiment-analysis', model='siebert/sentiment-roberta-large-english')" || echo "Failed to download model 2"
+    pipeline('sentiment-analysis', model='siebert/sentiment-roberta-large-english', use_auth_token='$HF_TOKEN')" || echo 'Failed to download model 2'
 
 # Expose port for FastAPI
 EXPOSE 7860