Update app.py

app.py

@@ -11,14 +11,14 @@ from tokenizers import Regex
 os.environ["TORCHINDUCTOR_DISABLE"] = "1"
 os.environ["USER"] = os.environ.get("USER", "appuser")
 
-
-# =====================================================
-# ✅ Safe Cache Setup
-# =====================================================
 CACHE_DIR = "/tmp/hf_cache"
+os.makedirs(CACHE_DIR, exist_ok=True)
 os.environ["HF_HOME"] = CACHE_DIR
 os.environ["TRANSFORMERS_CACHE"] = CACHE_DIR
-os.makedirs(CACHE_DIR, exist_ok=True)
+os.environ["HF_DATASETS_CACHE"] = CACHE_DIR
+os.environ["HF_HUB_CACHE"] = CACHE_DIR
+os.environ["TORCH_HOME"] = CACHE_DIR
+os.environ["XDG_CACHE_HOME"] = CACHE_DIR
 
 # =====================================================
 # ✅ Model Setup