Use hf_hub to download model at runtime; remove local checkpoint from Space app

app.py

@@ -4,10 +4,12 @@ import torch
 import gradio as gr
 
 from mmseg.apis import init_model, inference_model
+from huggingface_hub import hf_hub_download
 
-# Paths to model assets inside the repo
-CONFIG_PATH = "model/segformer_mit-b5_8xb1-160k_pre-cityscapes_seaicergb0-1024x1024.py"
-CHECKPOINT_PATH = "model/iter_160000.pth"
+# Source model repository on the Hub (avoid storing large files in the Space)
+REPO_ID = "triton7777/SeaIce"
+CONFIG_FILENAME = "segformer_mit-b5_8xb1-160k_pre-cityscapes_seaicergb0-1024x1024.py"
+CHECKPOINT_FILENAME = "iter_160000.pth"
 
 _model = None
 _palette = None
@@ -17,7 +19,29 @@ def _get_model():
     global _model, _palette
     if _model is None:
         device = "cuda" if torch.cuda.is_available() else "cpu"
-        _model = init_model(CONFIG_PATH, CHECKPOINT_PATH, device=device)
+        # Prefer local files if present (for local dev), otherwise download from Hub
+        local_cfg = os.path.join("model", CONFIG_FILENAME)
+        local_ckpt = os.path.join("model", CHECKPOINT_FILENAME)
+
+        if os.path.exists(local_cfg) and os.path.exists(local_ckpt):
+            cfg_path = local_cfg
+            ckpt_path = local_ckpt
+        else:
+            def _download_any(name: str):
+                candidates = [name, os.path.join("model", name)]
+                last_err = None
+                for cand in candidates:
+                    try:
+                        return hf_hub_download(repo_id=REPO_ID, filename=cand)
+                    except Exception as e:  # pragma: no cover (runtime fallback)
+                        last_err = e
+                        continue
+                raise last_err
+
+            cfg_path = _download_any(CONFIG_FILENAME)
+            ckpt_path = _download_any(CHECKPOINT_FILENAME)
+
+        _model = init_model(cfg_path, ckpt_path, device=device)
         meta = getattr(_model, "dataset_meta", None) or {}
         _palette = meta.get("palette", None)
         if _palette is None:

model/iter_160000.pth

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

requirements.txt

@@ -13,3 +13,4 @@ opencv-python-headless==4.8.1.78
 numpy==1.24.4
 gradio==4.12.0
 pillow>=9.4.0
+huggingface_hub>=0.20.0