Update Dockerfile

Dockerfile

@@ -31,18 +31,23 @@ RUN mkdir -p $HF_HOME && \
 # Copy application files
 COPY main.py .  
 
-# Set up Hugging Face token as a build argument (secured via environment variables)
+# Set up Hugging Face token as a build argument
 ARG HF_TOKEN
 ENV HF_TOKEN=${HF_TOKEN}
 
-# Pre-download models as the new user
+# Switch to the non-root user
 USER appuser
 
-RUN /app/venv/bin/python -c "from transformers import pipeline; \
-    pipeline('sentiment-analysis', model='Ehrii/sentiment', use_auth_token='$HF_TOKEN')" || echo 'Failed to download multilingual model'
+# Authenticate with Hugging Face and download models
+RUN /app/venv/bin/python -c "from huggingface_hub import login; \
+    login(token='$HF_TOKEN'); \
+    from transformers import pipeline; \
+    pipeline('sentiment-analysis', model='Ehrii/sentiment')" || echo 'Failed to download multilingual model'
 
-RUN /app/venv/bin/python -c "from transformers import pipeline; \
-    pipeline('sentiment-analysis', model='siebert/sentiment-roberta-large-english', use_auth_token='$HF_TOKEN')" || echo 'Failed to download English model'
+RUN /app/venv/bin/python -c "from huggingface_hub import login; \
+    login(token='$HF_TOKEN'); \
+    from transformers import pipeline; \
+    pipeline('sentiment-analysis', model='siebert/sentiment-roberta-large-english')" || echo 'Failed to download English model'
 
 # Expose FastAPI port
 EXPOSE 7860