Rajan Sharma commited on
Commit
d312515
·
verified ·
1 Parent(s): 7125686

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -3
app.py CHANGED
@@ -11,13 +11,30 @@ from settings import SNAPSHOT_PATH, PERSIST_CONTENT
11
  from audit_log import log_event, hash_summary
12
  from privacy import redact_text
13
 
14
- # ---------- Environment / cache ----------
 
15
  os.environ.setdefault("HF_HOME", "/data/.cache/huggingface")
16
  os.environ.setdefault("HF_HUB_CACHE", "/data/.cache/huggingface/hub")
 
 
 
 
 
 
17
  os.environ.setdefault("GRADIO_TEMP_DIR", "/data/gradio")
18
  os.environ.setdefault("GRADIO_CACHE_DIR", "/data/gradio")
19
- os.environ.pop("TRANSFORMERS_CACHE", None)
20
- for p in ["/data/.cache/huggingface/hub", "/data/gradio"]:
 
 
 
 
 
 
 
 
 
 
21
  try:
22
  os.makedirs(p, exist_ok=True)
23
  except Exception:
 
11
  from audit_log import log_event, hash_summary
12
  from privacy import redact_text
13
 
14
+ # ---------- Environment / cache (Spaces-safe, writable) ----------
15
+ # Hugging Face caches
16
  os.environ.setdefault("HF_HOME", "/data/.cache/huggingface")
17
  os.environ.setdefault("HF_HUB_CACHE", "/data/.cache/huggingface/hub")
18
+ os.environ.setdefault("TRANSFORMERS_CACHE", "/data/.cache/huggingface/transformers")
19
+
20
+ # SentenceTransformers cache (used by retriever.py)
21
+ os.environ.setdefault("SENTENCE_TRANSFORMERS_HOME", "/data/.cache/sentence-transformers")
22
+
23
+ # Gradio temp/cache
24
  os.environ.setdefault("GRADIO_TEMP_DIR", "/data/gradio")
25
  os.environ.setdefault("GRADIO_CACHE_DIR", "/data/gradio")
26
+
27
+ # Disable experimental xet transport; use stable transfer
28
+ os.environ.setdefault("HF_HUB_ENABLE_XET", "0")
29
+ os.environ.setdefault("HF_HUB_ENABLE_HF_TRANSFER", "1")
30
+
31
+ for p in [
32
+ "/data/.cache/huggingface",
33
+ "/data/.cache/huggingface/hub",
34
+ "/data/.cache/huggingface/transformers",
35
+ "/data/.cache/sentence-transformers",
36
+ "/data/gradio",
37
+ ]:
38
  try:
39
  os.makedirs(p, exist_ok=True)
40
  except Exception: