Update Dockerfile

Dockerfile

@@ -20,7 +20,7 @@ RUN mkdir -p /.cache/huggingface && \
     chmod -R 777 /.cache
 
 # Set environment variables for cache locations
-ENV TRANSFORMERS_CACHE="/.cache/huggingface"
+ENV HF_HOME="/.cache/huggingface"
 ENV TORCH_HOME="/.cache/torch"
 ENV SENTENCE_TRANSFORMERS_HOME="/.cache/sentence_transformers"
 
@@ -36,7 +36,8 @@ RUN pip install --no-cache-dir \
     uvicorn[standard] \
     pydantic \
     python-multipart \
-    huggingface_hub
+    huggingface_hub \
+    accelerate>=0.26.0
 
 # Copy the necessary files into the container
 COPY plant_data_chunks_and_embeddings.csv /app/plant_data_chunks_and_embeddings.csv