Update Dockerfile

Dockerfile

@@ -1,26 +1,31 @@
+# Use lightweight Python base image
 FROM python:3.9-slim
 
-# Set environment variables for Hugging Face cache
+# Set Hugging Face cache directory (replaces deprecated TRANSFORMERS_CACHE)
 ENV HF_HOME=/app/.cache/huggingface
 
+# Set working directory
 WORKDIR /app
 
-# Install dependencies
+# Create cache directory with open permissions
+RUN mkdir -p /app/.cache/huggingface && chmod -R 777 /app/.cache/huggingface
+
+# Install Python dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy your app source code
-COPY src/ ./src
-
-# Set environment variable for HF token (pass your real token at build/run time or use build args)
-ARG HF_TOKEN
-ENV HF_TOKEN=${HF_TOKEN}
+# Copy app files
+COPY . .
 
-# 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')"
+# Preload gated and public models using secret HF_TOKEN
+RUN python3 -c "import os; from transformers import pipeline; \
+  pipe = pipeline('summarization', model='mistralai/Mistral-7B-Instruct-v0.1', \
+  tokenizer='mistralai/Mistral-7B-Instruct-v0.1', use_auth_token=os.getenv('HF_TOKEN'))" && \
+  python3 -c "from transformers import pipeline; \
+  pipeline('sentiment-analysis', model='distilbert-base-uncased-finetuned-sst-2-english')"
 
-# Expose port for Streamlit
-EXPOSE 8501
+# Expose Streamlit default port
+EXPOSE 7860
 
-# Run your Streamlit app
-CMD ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]
+# Run Streamlit app
+CMD ["streamlit", "run", "src/streamlit_app.py", "--server.port=7860", "--server.address=0.0.0.0"]