Update app.py

app.py

@@ -12,8 +12,14 @@ import tempfile
 from pathlib import Path
 import pydicom
 
-CACHE_DIR = Path(tempfile.gettempdir()) / "huggingface_cache"
-os.environ["TRANSFORMERS_CACHE"] = str(CACHE_DIR)
+CACHE_DIR = Path("/huggingface/cache")
+HF_CACHE_DIR = Path("/huggingface/cache")
+
+os.environ["TRANSFORMERS_CACHE"] = str(HF_CACHE_DIR)
+os.environ["HF_HOME"] = str(HF_CACHE_DIR.parent)
+os.environ["HF_HUB_CACHE"] = str(HF_CACHE_DIR)
+os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
+
 CACHE_DIR.mkdir(parents=True, exist_ok=True)
 
 app = FastAPI()