Update Dockerfile

Dockerfile

@@ -1,11 +1,23 @@
-FROM python:3.10
-
-WORKDIR /code
-
-COPY ./requirements.txt /code/requirements.txt
-
-RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
-
-COPY . /code
-
-CMD ["uvicorn", "fapi:app", "--host", "0.0.0.0", "--port", "7860"]
+FROM python:3.10
+
+WORKDIR /code
+
+# Create cache directories with proper permissions
+RUN mkdir -p /tmp/cache/huggingface && \
+    chmod 777 /tmp/cache/huggingface
+
+# Set environment variables for cache directories
+ENV HF_HOME=/tmp/cache/huggingface
+ENV TRANSFORMERS_CACHE=/tmp/cache/huggingface
+ENV SENTENCE_TRANSFORMERS_HOME=/tmp/cache/huggingface
+
+COPY ./requirements.txt /code/requirements.txt
+
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+# Pre-download the sentence transformer model during build
+RUN python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('all-MiniLM-L6-v2')"
+
+COPY . /code
+
+CMD ["uvicorn", "fapi:app", "--host", "0.0.0.0", "--port", "7860"]