Update Dockerfile

Dockerfile

@@ -31,26 +31,21 @@ RUN mkdir -p $HF_HOME && \
 # Copy application files
 COPY main.py .  
 
-# Set up Hugging Face token as a build argument
+# Set up Hugging Face token as a build argument (secured via environment variables)
 ARG HF_TOKEN
 ENV HF_TOKEN=${HF_TOKEN}
 
-# Switch to the non-root user
+# Pre-download models as the new user
 USER appuser
 
-# 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='Ehrii/sentiment', use_auth_token='$HF_TOKEN')" || echo 'Failed to download multilingual 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'
+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'
 
 # Expose FastAPI port
 EXPOSE 7860
 
 # Run FastAPI server using the virtual environment
-CMD ["/app/venv/bin/uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["/app/venv/bin/uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]