Update Dockerfile

Dockerfile

@@ -3,6 +3,13 @@ FROM python:3.10
 # Set the working directory
 WORKDIR /app
 
+# Set environment variables for Hugging Face cache location
+ENV TRANSFORMERS_CACHE=/app/models_cache
+ENV HF_HOME=/app/models_cache
+ENV HF_DATASETS_CACHE=/app/models_cache
+ENV SENTENCE_TRANSFORMERS_HOME=/app/models_cache
+ENV XDG_CACHE_HOME=/app/models_cache
+
 # Copy and install dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
@@ -17,4 +24,4 @@ RUN mkdir -p /app/models_cache && chmod -R 777 /app/models_cache
 EXPOSE 7860
 
 # Run the FastAPI application with Uvicorn
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]