Update Dockerfile

Dockerfile

@@ -11,18 +11,21 @@ RUN apt-get update && apt-get install -y \
     && rm -rf /var/lib/apt/lists/*
 
 # Create cache directory with proper permissions
-RUN mkdir -p /root/.cache/huggingface && \
-    chmod -R 777 /root/.cache
+RUN mkdir -p /.cache/huggingface && \
+    chmod -R 777 /.cache
 
 # Set environment variables
-ENV TRANSFORMERS_CACHE=/root/.cache/huggingface
-ENV HF_HOME=/root/.cache/huggingface
+ENV TRANSFORMERS_CACHE=/.cache/huggingface
+ENV HF_HOME=/.cache/huggingface
+ENV TORCH_HOME=/.cache/torch
+ENV XDG_CACHE_HOME=/.cache
 
 # Copy requirements first for better caching
 COPY requirements.txt .
 
 # Install Python dependencies
-RUN pip install --no-cache-dir -r requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt && \
+    python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('jinaai/jina-embeddings-v3', trust_remote_code=True)"
 
 # Copy the rest of the application
 COPY . .