Update app.py

app.py

@@ -1,21 +1,27 @@
-import os
-
-# ============================================================
-# ✅ FIX FOR "state dict corrupted" ON SPACES (Xet downloads)
-# ============================================================
-# Disable hf-xet usage (forces download via LFS bridge instead).
-# HF docs: HF_HUB_DISABLE_XET disables using hf-xet. :contentReference[oaicite:2]{index=2}
-os.environ["HF_HUB_DISABLE_XET"] = "1"
-
-# Optional: place HF cache in a writable/temp location (Spaces friendly)
-# You can comment this out if you prefer default cache locations.
-os.environ.setdefault("HF_HOME", "/tmp/hf")
-os.environ.setdefault("HUGGINGFACE_HUB_CACHE", "/tmp/hf/hub")
-os.environ.setdefault("TRANSFORMERS_CACHE", "/tmp/hf/transformers")
-
-# (Optional) reduce parallelism issues
-os.environ.setdefault("TOKENIZERS_PARALLELISM", "false")
-
+import os, shutil, glob
+
+# Put HF caches somewhere "fresh" (avoid reusing an old corrupt cache)
+os.environ["HF_HOME"] = "/tmp/hf"
+os.environ["HUGGINGFACE_HUB_CACHE"] = "/tmp/hf/hub"
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf/transformers"
+os.environ["HF_HUB_DISABLE_XET"] = "1"     # also avoids xet-related partial downloads
+os.environ["TOKENIZERS_PARALLELISM"] = "false"
+
+def wipe_model_cache(model_id: str):
+    safe = model_id.replace("/", "--")
+    paths = [
+        f"/tmp/hf/hub/models--{safe}",
+        f"/tmp/hf/transformers/models--{safe}",
+        # also wipe common defaults in case something else wrote there
+        os.path.expanduser(f"~/.cache/huggingface/hub/models--{safe}"),
+        os.path.expanduser(f"~/.cache/huggingface/transformers/models--{safe}"),
+    ]
+    for p in paths:
+        if os.path.exists(p):
+            shutil.rmtree(p, ignore_errors=True)
+
+# wipe the specific model cache on startup
+wipe_model_cache("desklib/ai-text-detector-v1.01")
 
 import re
 import shutil