fix: .cache issue

gradio_app.py

@@ -23,13 +23,16 @@ from notebook_lm_kokoro import (
     KPipeline,
 )
 
-os.environ["HOME"] = "/tmp/home"  # Prevent fallback to /root
-os.makedirs("/tmp/home", exist_ok=True)
-
-
+# Set cache dirs BEFORE importing torch, transformers, or moshi
+os.environ["HF_HOME"] = "/tmp/huggingface"
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface/transformers"
+os.environ["XDG_CACHE_HOME"] = "/tmp/huggingface"
 os.environ["TORCH_HOME"] = "/tmp/torch"
 os.environ["MOSHI_CACHE_DIR"] = "/tmp/moshi"
-os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface/transformers"
+
+# Explicitly override ~/.cache
+os.environ["HOME"] = "/tmp/home"
+os.makedirs("/tmp/home", exist_ok=True)
 import gradio as gr
 
 warnings.filterwarnings("ignore")

notebook_lm_kokoro.py

@@ -38,12 +38,17 @@ openai.api_key = os.getenv("OPENAI_API_KEY")
 # For OpenRouter compatibility, set the API base if provided.
 openai.api_base = os.getenv("OPENROUTER_API_BASE", "https://api.openai.com/v1")
 
+# Set cache dirs BEFORE importing torch, transformers, or moshi
 os.environ["HF_HOME"] = "/tmp/huggingface"
 os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface/transformers"
 os.environ["XDG_CACHE_HOME"] = "/tmp/huggingface"
 os.environ["TORCH_HOME"] = "/tmp/torch"
 os.environ["MOSHI_CACHE_DIR"] = "/tmp/moshi"
 
+# Explicitly override ~/.cache
+os.environ["HOME"] = "/tmp/home"
+os.makedirs("/tmp/home", exist_ok=True)
+
 pdf = "1706.03762v7.pdf"