Update Dockerfile

Dockerfile

@@ -1,29 +1,26 @@
-# Use lightweight Python 3.9 image
 FROM python:3.9-slim
 
+# Set environment variables for Hugging Face cache
+ENV HF_HOME=/app/.cache/huggingface
+
 WORKDIR /app
 
 # Install dependencies
-RUN apt-get update && apt-get install -y \
-    git curl build-essential && rm -rf /var/lib/apt/lists/*
-
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-COPY src/ ./src/
+# Copy your app source code
+COPY src/ ./src
 
-# Set Hugging Face cache directory
-ENV HF_HOME=/app/.cache/huggingface
-ENV TRANSFORMERS_CACHE=/app/.cache/huggingface/transformers
-
-# Accept token securely at build time
+# Set environment variable for HF token (pass your real token at build/run time or use build args)
 ARG HF_TOKEN
-RUN echo "machine huggingface.co\n  login ${HF_TOKEN}" > /root/.netrc && chmod 600 /root/.netrc
+ENV HF_TOKEN=${HF_TOKEN}
 
-# Skip pre-download here (let Streamlit download at runtime securely)
-# If you really want pre-download, use this ONLY if token works
-# RUN python3 -c "from transformers import pipeline; pipeline('summarization', model='mistralai/Mistral-7B-Instruct-v0.1', use_auth_token=True)"
+# Optional: pre-download models (skip if gated or use runtime download)
+# RUN python3 -c "from transformers import pipeline; pipeline('sentiment-analysis', model='distilbert-base-uncased-finetuned-sst-2-english')"
 
-# Port and entrypoint
+# Expose port for Streamlit
 EXPOSE 8501
-ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]
+
+# Run your Streamlit app
+CMD ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]