App: place caches in persistent storage

app.py

@@ -6,6 +6,11 @@ import sys
 os.environ["TORCH_COMPILE_DISABLE"] = "1"
 os.environ["TORCHDYNAMO_DISABLE"] = "1"
 
+# Place caches in persistent storage
+os.environ["HF_HOME"] = /data/.huggingface
+os.environ["HF_HUB_CACHE"] = /data/.cache/huggingface/hub
+os.environ["HF_DATASETS_CACHE"] = /data/.cache/huggingface/datasets
+
 # Install xformers for memory-efficient attention
 subprocess.run([sys.executable, "-m", "pip", "install", "xformers==0.0.32.post2", "--no-build-isolation"], check=False)