Update Dockerfile

Dockerfile

@@ -1,22 +1,21 @@
-# Use the official lightweight Python image
 FROM python:3.10-slim
 
-# Set working directory
 WORKDIR /app
 
-# Copy source files
+# Copy files
 COPY ./src /app/src
 COPY requirements.txt .
 
-# Create a writable cache directory for transformers
+# Create a safe cache directory
 RUN mkdir -p /app/cache
 
 # Install dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Set environment variables
+# Set environment variables for safe caching
+ENV HF_HOME=/app/cache
 ENV TRANSFORMERS_CACHE=/app/cache
 ENV STREAMLIT_SERVER_HEADLESS=true
 
-# Run the app using Streamlit
+# Run the app
 CMD ["streamlit", "run", "src/streamlit_app.py", "--server.port=7860", "--server.address=0.0.0.0"]