Update Dockerfile

Dockerfile

@@ -1,7 +1,8 @@
-FROM python:3.9-slim
+FROM python:3.11-slim
 
 WORKDIR /app
 
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
     build-essential \
     curl \
@@ -9,10 +10,21 @@ RUN apt-get update && apt-get install -y \
     git \
     && rm -rf /var/lib/apt/lists/*
 
+# Set environment variables for cache directories
+ENV STREAMLIT_HOME=/app/.streamlit
+ENV HF_HOME=/app/.cache/huggingface
+ENV TRANSFORMERS_CACHE=/app/.cache/transformers
+ENV TORCH_HOME=/app/.cache/torch
+
+# Copy requirements and install Python dependencies
 COPY requirements.txt ./
+RUN pip3 install -r requirements.txt
+
+# Copy source code
 COPY src/ ./src/
 
-RUN pip3 install -r requirements.txt
+# Create necessary directories with proper permissions
+RUN mkdir -p /app/.streamlit /app/.cache/huggingface /app/.cache/transformers /app/.cache/torch /app/data /app/chroma
 
 EXPOSE 8501