Update Dockerfile

Dockerfile

@@ -1,15 +1,16 @@
 # Use lightweight python 3.9 image
 FROM python:3.9-slim
+
 WORKDIR /app
 
-# Install system dependencies
+# Install system deps
 RUN apt-get update && apt-get install -y \
     build-essential \
     git \
-    curl \
-    && rm -rf /var/lib/apt/lists/*
+    curl && \
+    rm -rf /var/lib/apt/lists/*
 
-# Set environment variables
+# Set Hugging Face cache path
 ENV HF_HOME=/app/model_cache
 ENV STREAMLIT_CONFIG_DIR=/app/.streamlit
 
@@ -18,24 +19,19 @@ COPY requirements.txt ./
 COPY .streamlit/ .streamlit/
 COPY src/ ./src/
 
-# Install Python dependencies
+# Install Python deps
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Create cache and config directories with proper permissions
+# Create cache and config dirs with permissions
 RUN mkdir -p /app/model_cache && chmod -R 777 /app/model_cache
 RUN mkdir -p /app/.streamlit && chmod -R 777 /app/.streamlit
 
-# Pre-download all models to cache (this caches in /app/model_cache)
-RUN python -c "from transformers import pipeline; pipeline('zero-shot-classification', model='facebook/bart-large-mnli', use_safetensors=True)" && \
-    python -c "from transformers import pipeline; pipeline('sentiment-analysis', model='distilbert-base-uncased-finetuned-sst-2-english', use_safetensors=True)" && \
-    python -c "from transformers import pipeline; pipeline('summarization', model='sshleifer/distilbart-cnn-6-6', use_safetensors=True)" && \
+# Pre-download models to cache
+RUN python -c "from transformers import pipeline; pipeline('zero-shot-classification', model='facebook/bart-large-mnli')" && \
+    python -c "from transformers import pipeline; pipeline('sentiment-analysis', model='distilbert-base-uncased-finetuned-sst-2-english')" && \
+    python -c "from transformers import pipeline; pipeline('summarization', model='sshleifer/distilbart-cnn-6-6')" && \
     python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('all-MiniLM-L6-v2')"
 
-# Expose Streamlit port
 EXPOSE 8501
-
-# Healthcheck to verify Streamlit is running
 HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health || exit 1
-
-# Entrypoint to launch Streamlit app
 ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]