a

Dockerfile

@@ -6,11 +6,11 @@ COPY . .
 
 # Set the working directory to /
 WORKDIR /
-ENV HF_HOME=/.cache/huggingface/hub
-ENV TRANSFORMERS_CACHE=/.cache/huggingface/hub
+ENV HF_HOME=/app/huggingface_cache 
 # Set environment variables
 ENV HUGGINGFACE_TOKEN=""
-RUN mkdir -p /.cache/huggingface/hub
+RUN mkdir -p /app/huggingface_cache && chmod -R 777 /app/huggingface_cache
+
 
 # Install requirements.txt 
 RUN pip install --no-cache-dir --upgrade -r /requirements.txt

app.py

@@ -6,6 +6,10 @@ from huggingface_hub import login
 import torch
 import os
 
+
+os.environ["HF_HOME"] = "/app/huggingface_cache"
+# Ensure the cache directory exists
+os.makedirs("/app/huggingface_cache", exist_ok=True)
 # Initialize FastAPI app
 app = FastAPI()
 hf_token = os.environ.get("HUGGINGFACE_TOKEN")