Upload folder using huggingface_hub

Dockerfile

@@ -12,6 +12,11 @@ RUN apt-get update && apt-get install -y \
     curl \
  && rm -rf /var/lib/apt/lists/*
 
+ENV HF_HOME=/tmp/huggingface
+ENV TRANSFORMERS_CACHE=/tmp/huggingface/transformers
+ENV HF_DATASETS_CACHE=/tmp/huggingface/datasets
+ENV DIFFUSERS_CACHE=/tmp/huggingface/diffusers
+
 # Copy files
 COPY . .
 

app.py

@@ -24,14 +24,17 @@ templates = Jinja2Templates(directory="templates")
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
-os.environ["HF_HOME"] = "/tmp/hf_home"
-os.environ["HF_DATASETS_CACHE"] = "/tmp/hf_datasets"
+os.environ["HF_HOME"] = "/tmp/huggingface"
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface/transformers"
+os.environ["HF_DATASETS_CACHE"] = "/tmp/huggingface/datasets"
+os.environ["DIFFUSERS_CACHE"] = "/tmp/huggingface/diffusers"
 
 pipe = StableDiffusionPipeline.from_pretrained(
     "runwayml/stable-diffusion-v1-5",
+    cache_dir="/tmp/huggingface",
     torch_dtype=torch.float16 if device == "cuda" else torch.float32
 )
+
 pipe.to(device)