Upload README.md with huggingface_hub

README.md

@@ -0,0 +1,38 @@
+---
+license: cc-by-nc-4.0
+tags:
+  - pathology
+  - cell-classification
+  - pytorch
+  - torch-export
+---
+
+# MetPredict-cell-reclassifier
+
+Per-cell MLP classifier. Maps **histoplus 768-dim cell features** to
+softmax probabilities over: `['background', 'tumor']`.
+
+Exported with `torch.export` (batch dim: `dynamic`, feature dim: `768`).
+The exported program wraps `MLP + softmax`, so the output is probabilities
+directly — no model class required at load time.
+
+## Usage
+
+```python
+from huggingface_hub import hf_hub_download
+import json, torch
+
+path = hf_hub_download("RendeiroLab/MetPredict-cell-reclassifier", "model.pt2")
+labels = json.loads(open(hf_hub_download("RendeiroLab/MetPredict-cell-reclassifier", "labels.json")).read())["labels"]
+
+m = torch.export.load(path).module()
+probs = m(torch.randn(N, 768))       # (N, 2), softmaxed
+pred_idx = probs.argmax(-1)
+pred_class = [labels[i] for i in pred_idx.tolist()]
+```
+
+## Files
+
+- `model.pt2` — `torch.export` program (MLP + softmax)
+- `labels.json` — class index → name map, plus `in_dim` / `n_classes` / `batch`
+- `README.md` — this file