Default demo repo_types to model; clarify errors

app.py

@@ -29,6 +29,9 @@ HUB_REPO_ID = os.getenv("LWM_SPECTRO_DEMO_REPO_ID", "wi-lab/lwm-spectro")
 HUB_REVISION = os.getenv("LWM_SPECTRO_DEMO_REVISION")  # optional git sha / tag / branch
 HUB_DEMO_DATA_FILENAME = os.getenv("LWM_SPECTRO_DEMO_DATA_FILENAME", "demo_data.pt")
 HUB_MOE_DATA_FILENAME = os.getenv("LWM_SPECTRO_MOE_DATA_FILENAME", "demo_data_moe.pt")
+HUB_REPO_TYPES = tuple(
+    t.strip() for t in os.getenv("LWM_SPECTRO_DEMO_REPO_TYPES", "model").split(",") if t.strip()
+)
 
 
 def _get_hf_token() -> str | None:
@@ -207,9 +210,9 @@ def _ensure_local_file(local_path: Path, hub_filename: str) -> Optional[Path]:
     # even when HF_TOKEN env var is not explicitly set).
     token = HF_TOKEN or True
 
-    # Try multiple repo types because the artifact may live under a model or dataset repo.
+    # Try configured repo types (default: model). This Space historically used a model repo.
     last_exc: Exception | None = None
-    for repo_type in ("model", "dataset"):
+    for repo_type in HUB_REPO_TYPES:
         try:
             cached = hf_hub_download(
                 repo_id=HUB_REPO_ID,
@@ -243,7 +246,8 @@ def _ensure_local_file(local_path: Path, hub_filename: str) -> Optional[Path]:
             err = err[:240] + "..."
         LAST_DEMO_DOWNLOAD_ERROR = err
         print(
-            f"[WARN] Could not download {hub_filename} from Hub ({last_exc}) or Space repo ({exc}); continuing without it."
+            f"[WARN] Could not download {hub_filename} from Hub (repo_id={HUB_REPO_ID}, repo_types={HUB_REPO_TYPES}, revision={HUB_REVISION or 'main'}: {last_exc}) "
+            f"or Space repo ({exc}); continuing without it."
         )
         return None
 
@@ -921,7 +925,7 @@ DATASET_STATUS = (
     f"MoE embeddings: {'yes' if has_moe_embeddings else 'no'} | "
     f"HF token detected: {'yes' if HF_TOKEN else 'no'} | "
     f"Synthetic fallback: {'yes' if USING_SYNTHETIC_DATA else 'no'} | "
-    f"Demo repo: {HUB_REPO_ID}@{HUB_REVISION or 'main'}"
+    f"Demo repo: {HUB_REPO_ID}@{HUB_REVISION or 'main'} ({','.join(HUB_REPO_TYPES)})"
 )
 
 if LAST_DEMO_DOWNLOAD_ERROR: