Space: auto-download model from HF model repo

app.py

@@ -50,15 +50,39 @@ NUM_PAIRS = None
 BATCH_RESULTS = None
 BATCH_SHAP_CACHE = {}
 
+MODEL_FILENAME = "pairwise_onehot_model_20260105_200141.pt"
+MODEL_REPO_ID = os.getenv("MIRBIND2_MODEL_REPO", "dimostzim/mirbind2-weights")
+
+
+def resolve_model_path():
+    """Resolve model path locally, or download from Hugging Face Hub if missing."""
+    local_model_path = Path(__file__).parent / "miRBind_2.0-main" / "models" / MODEL_FILENAME
+    if local_model_path.exists():
+        return local_model_path
+
+    print(f"Local model not found at {local_model_path}")
+    print(f"Downloading model from Hugging Face Hub repo: {MODEL_REPO_ID}")
+
+    try:
+        from huggingface_hub import hf_hub_download
+    except ImportError as exc:
+        raise FileNotFoundError(
+            "Model file missing locally and huggingface_hub is not installed."
+        ) from exc
+
+    downloaded_path = hf_hub_download(
+        repo_id=MODEL_REPO_ID,
+        filename=MODEL_FILENAME,
+        repo_type="model",
+    )
+    return Path(downloaded_path)
+
 
 def load_pretrained_model():
     """Load the pre-trained model once at startup."""
     global MODEL, MODEL_PARAMS, DEVICE, PAIR_TO_INDEX, NUM_PAIRS
 
-    model_path = Path(__file__).parent / "miRBind_2.0-main" / "models" / "pairwise_onehot_model_20260105_200141.pt"
-
-    if not model_path.exists():
-        raise FileNotFoundError(f"Model not found at {model_path}")
+    model_path = resolve_model_path()
 
     DEVICE = get_device()
     PAIR_TO_INDEX = get_pair_to_index()

requirements.txt

@@ -8,3 +8,4 @@ matplotlib>=3.7
 Pillow>=10.0
 captum>=0.7.0
 scikit-learn>=1.3.0
+huggingface_hub>=0.30

upload_model_to_hf_model.sh

@@ -0,0 +1,27 @@
+#!/usr/bin/env bash
+set -euo pipefail
+
+# Upload miRBind2 model weights to a Hugging Face Model repo.
+# Usage:
+#   ./upload_model_to_hf_model.sh
+#   MODEL_REPO_ID=dimostzim/mirbind2-weights ./upload_model_to_hf_model.sh
+#   ./upload_model_to_hf_model.sh /path/to/local.pt pairwise_onehot_model_20260105_200141.pt
+
+MODEL_REPO_ID="${MODEL_REPO_ID:-dimostzim/mirbind2-weights}"
+LOCAL_MODEL_PATH="${1:-$HOME/webapp_package/miRBind_2.0-main/models/pairwise_onehot_model_20260105_200141.pt}"
+REMOTE_MODEL_PATH="${2:-pairwise_onehot_model_20260105_200141.pt}"
+
+if [[ ! -f "$LOCAL_MODEL_PATH" ]]; then
+  echo "Local model file not found:"
+  echo "  $LOCAL_MODEL_PATH"
+  exit 1
+fi
+
+if command -v hf >/dev/null 2>&1; then
+  hf upload "$MODEL_REPO_ID" "$LOCAL_MODEL_PATH" "$REMOTE_MODEL_PATH" --repo-type model
+else
+  python -m pip install -U huggingface_hub
+  python -m huggingface_hub upload "$MODEL_REPO_ID" "$LOCAL_MODEL_PATH" "$REMOTE_MODEL_PATH" --repo-type model
+fi
+
+echo "Model uploaded to: https://huggingface.co/$MODEL_REPO_ID"