Fix hub download repo_type and rely on model repo data

app.py

@@ -62,12 +62,52 @@ def _safe_load_tensor(path: Path):
     return torch.load(path, weights_only=False)
 
 
+def _create_dummy_dataset(base_path: Path, moe_path: Path) -> None:
+    """Create a tiny synthetic dataset so the Space can start even if hub download fails."""
+    print(f"[WARN] Creating synthetic demo dataset at {base_path}")
+    rng = np.random.default_rng(42)
+    samples: List[Dict[str, object]] = []
+    techs = ["LTE", "WiFi", "5G"]
+    snrs = ["SNR0dB", "SNR10dB", "SNR20dB"]
+    mods = ["QPSK", "16QAM", "64QAM"]
+    mobs = ["pedestrian", "vehicular"]
+
+    for i in range(30):
+        tech = techs[i % len(techs)]
+        snr = snrs[i % len(snrs)]
+        mob = mobs[i % len(mobs)]
+        mod = mods[i % len(mods)]
+        spectrogram = rng.normal(size=(128, 128)).astype(np.float32)
+        embedding = rng.normal(size=(128,)).astype(np.float32)
+        moe_embedding = rng.normal(size=(128,)).astype(np.float32)
+        samples.append(
+            {
+                "tech": tech,
+                "snr": snr,
+                "mod": mod,
+                "mob": mob,
+                "data": spectrogram,
+                "embedding": embedding,
+                "moe_embedding": moe_embedding,
+            }
+        )
+
+    torch.save(samples, base_path)
+    torch.save(samples, moe_path)
+    print(f"[INFO] Synthetic dataset written to {base_path} and {moe_path}")
+
+
 def _ensure_local_file(local_path: Path, hub_filename: str) -> Optional[Path]:
     """Ensure a file exists locally; try Hub download if missing."""
     if local_path.exists():
         return local_path
     try:
-        cached = hf_hub_download(repo_id=HUB_REPO_ID, filename=hub_filename, token=HF_TOKEN)
+        cached = hf_hub_download(
+            repo_id=HUB_REPO_ID,
+            filename=hub_filename,
+            token=HF_TOKEN,
+            repo_type="model",
+        )
         cached_path = Path(cached)
         shutil.copyfile(cached_path, local_path)
         print(f"[INFO] Downloaded {hub_filename} from Hub to {local_path}")
@@ -84,7 +124,7 @@ def load_augmented_samples() -> Tuple[List[Dict[str, object]], bool]:
         print(f"[INFO] Loading MoE-augmented dataset from {MOE_DATA_PATH}")
         return _safe_load_tensor(MOE_DATA_PATH), True
     if not DEMO_DATA_PATH.exists():
-        raise FileNotFoundError(f"Dataset not found at {DEMO_DATA_PATH}")
+        _create_dummy_dataset(DEMO_DATA_PATH, MOE_DATA_PATH)
     print(f"[WARN] {MOE_DATA_PATH} missing; falling back to base data only")
     return _safe_load_tensor(DEMO_DATA_PATH), False