Update app.py

app.py

@@ -1,16 +1,24 @@
+import os
+
+# Set HF cache dirs to writable paths before importing HF libraries
+os.environ["HF_HOME"] = "/app/.cache/huggingface"
+os.environ["XDG_CACHE_HOME"] = "/app/.cache"
+
+# Make sure cache folders exist
+os.makedirs(os.environ["HF_HOME"], exist_ok=True)
+os.makedirs(os.environ["XDG_CACHE_HOME"], exist_ok=True)
+
+# Now import other libs
 from fastapi import FastAPI, BackgroundTasks
 import uuid
-import os
 from huggingface_hub import snapshot_download
 from flux_train import build_job
 import sys
 
-# Add ai-toolkit to sys.path for toolkit imports
 sys.path.append("/app/ai-toolkit")
-
 from toolkit.job import run_job
-import logging
-import io
+
+
 
 log_stream = io.StringIO()
 logger = logging.getLogger("lora_training")