Added local model weights

app.py

@@ -1,4 +1,5 @@
 import os
+import tarfile
 import tempfile
 from pathlib import Path
 from typing import List, Tuple
@@ -16,6 +17,7 @@ HF_BT_REPO = os.environ.get("HF_BT_REPO", "mschuh/TwinBooster-bt")
 
 # Store models where TwinBooster looks for them by default (~/.cache/twinbooster)
 MODEL_DIR = Path.home() / ".cache" / "twinbooster"
+WEIGHTS_SRC = Path(__file__).parent / "weights"
 
 
 def ensure_models() -> str:
@@ -40,11 +42,26 @@ def ensure_models() -> str:
         except Exception:
             return False
 
+    def extract_local(archive: Path, subdir: str) -> bool:
+        target = MODEL_DIR / subdir
+        if target.exists() or not archive.exists():
+            return False
+        target.mkdir(parents=True, exist_ok=True)
+        with tarfile.open(archive, "r:*") as tf:
+            tf.extractall(target)
+        return True
+
     if grab(HF_TEXT_REPO, "PubChemDeBERTa-augmented"):
         downloaded.append(HF_TEXT_REPO)
-    if HF_LGBM_REPO and grab(HF_LGBM_REPO, "lgbm_model"):
+
+    if extract_local(WEIGHTS_SRC / "lgbm_model.tar.xz", "lgbm_model"):
+        downloaded.append("local lgbm_model.tar.xz")
+    elif HF_LGBM_REPO and grab(HF_LGBM_REPO, "lgbm_model"):
         downloaded.append(HF_LGBM_REPO)
-    if HF_BT_REPO and grab(HF_BT_REPO, "bt_model"):
+
+    if extract_local(WEIGHTS_SRC / "bt_model.tar.xz", "bt_model"):
+        downloaded.append("local bt_model.tar.xz")
+    elif HF_BT_REPO and grab(HF_BT_REPO, "bt_model"):
         downloaded.append(HF_BT_REPO)
 
     # Ensure any missing pieces are resolved via package helper (will skip if already present)

weights/bt_model.tar.xz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:70607f078e00861c92e36c12a873a39e7e54842285653209ed81ac02337662b0
+size 466955972

weights/lgbm_model.tar.xz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e599e171a9cba778b9d858e283ee874addf9513ee97765b2890c729ac59a6bd3
+size 23041344