Upload folder using huggingface_hub

app.py

@@ -23,6 +23,11 @@ app = FastAPI()
 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"
+
 pipe = StableDiffusionPipeline.from_pretrained(
     "runwayml/stable-diffusion-v1-5",
     torch_dtype=torch.float16 if device == "cuda" else torch.float32