add SENTENCE_TRANSFORMERS_HOME to dockerfile

Dockerfile

@@ -10,6 +10,8 @@ RUN pip install -r requirements.txt
 # Copy the rest of the application code to the working directory
 COPY . .
 
+ENV SENTENCE_TRANSFORMERS_HOME ./.cache
+
 # Expose the port FastAPI will run on
 EXPOSE 7860
 

run.py

@@ -6,6 +6,7 @@ from src.prompt_loader import PromptLoader
 import os
 
 os.environ["SENTENCE_TRANSFORMERS_HOME"] = "./.cache"
+os.environ["HF_HOME"] = "./.cache"
 
 # Constants
 SEED = 42