Refresh README for current MLX and OpenMedKit instructions

README.md

@@ -16,90 +16,107 @@ tags:
 
 # OpenMed-PII-mLiteClinical-Base-135M-v1 for OpenMed MLX
 
-This repository contains a **private MLX packaging** of [`OpenMed/OpenMed-PII-mLiteClinical-Base-135M-v1`](https://huggingface.co/OpenMed/OpenMed-PII-mLiteClinical-Base-135M-v1) for Apple Silicon inference with [**OpenMed**](https://github.com/maziyarpanahi/openmed).
+This repository contains an MLX packaging of [`OpenMed/OpenMed-PII-mLiteClinical-Base-135M-v1`](https://huggingface.co/OpenMed/OpenMed-PII-mLiteClinical-Base-135M-v1) for Apple Silicon inference with [OpenMed](https://github.com/maziyarpanahi/openmed).
 
-[OpenMed](https://github.com/maziyarpanahi/openmed) is the main product experience:
+## At a Glance
 
-- Install the Python package with `pip install openmed`
-- Enable Apple Silicon acceleration with `pip install "openmed[mlx]"`
-- Run the same OpenMed API you already use, now backed by MLX on macOS
-- For Apple apps, use **OpenMedKit** from the same GitHub repository with a CoreML bundle
+- **Source checkpoint:** [`OpenMed/OpenMed-PII-mLiteClinical-Base-135M-v1`](https://huggingface.co/OpenMed/OpenMed-PII-mLiteClinical-Base-135M-v1)
+- **Model family:** `distilbert` (`DistilBertForTokenClassification`)
+- **Primary language hint:** English (`en`)
+- **Artifact layout:** legacy-compatible MLX (`config.json`, `id2label.json`, MLX weight files)
+- **Python MLX:** supported through `openmed[mlx]` on Apple Silicon Macs
+- **Swift MLX:** supported today in OpenMedKit on Apple Silicon macOS and real iPhone/iPad hardware
 
-This MLX repo is meant to pair with:
+## Python Quick Start
 
-- OpenMed GitHub: [https://github.com/maziyarpanahi/openmed](https://github.com/maziyarpanahi/openmed)
-- OpenMed website: [https://openmed.life](https://openmed.life)
-- Source checkpoint: [`OpenMed/OpenMed-PII-mLiteClinical-Base-135M-v1`](https://huggingface.co/OpenMed/OpenMed-PII-mLiteClinical-Base-135M-v1)
-
-## Quick Start
-
-### Python
+Use the standard OpenMed API if you want OpenMed to choose the right runtime automatically:
 
 ```bash
-pip install openmed
 pip install "openmed[mlx]"
 ```
 
 ```python
-from openmed import analyze_text
-from openmed.core.config import OpenMedConfig
+from openmed import extract_pii
 
-result = analyze_text(
-    "Patient John Doe, DOB 1990-05-15, SSN 123-45-6789",
+text = "<your clinical note here>"
+result = extract_pii(
+    text,
     model_name="OpenMed/OpenMed-PII-mLiteClinical-Base-135M-v1",
-    config=OpenMedConfig(backend="mlx"),
+    use_smart_merging=True,
 )
 
 for entity in result.entities:
     print(entity.label, entity.text, round(entity.confidence, 4))
 ```
 
-You can use the source model ID directly through OpenMed, or download this private MLX packaging from [`OpenMed/OpenMed-PII-mLiteClinical-Base-135M-v1-mlx`](https://huggingface.co/OpenMed/OpenMed-PII-mLiteClinical-Base-135M-v1-mlx) for a preconverted path.
+On Apple Silicon, OpenMed auto-selects MLX when `openmed[mlx]` is installed. On other systems it falls back to the Hugging Face / PyTorch backend.
+
+## Use This Preconverted MLX Repo Directly
 
-### Swift
+If you want to use this MLX snapshot explicitly, download it locally and point OpenMed at the directory:
 
-Use Swift with **OpenMedKit**, not with MLX weight files directly.
+```bash
+pip install "openmed[mlx]"
+hf download OpenMed/OpenMed-PII-mLiteClinical-Base-135M-v1-mlx --local-dir ./OpenMed-PII-mLiteClinical-Base-135M-v1-mlx
+```
 
-1. Open Xcode and go to **File > Add Package Dependencies...**
-2. Paste the OpenMed repository URL:
-   `https://github.com/maziyarpanahi/openmed`
-3. Choose the package product **OpenMedKit** from the repository.
-4. Add a compatible CoreML model bundle plus `id2label.json` to your app target.
+If this repo is still private in your environment, authenticate first with `hf auth login` or set `HF_TOKEN`.
 
-After that, import OpenMedKit in Swift:
+```python
+from openmed import extract_pii
+from openmed.core import OpenMedConfig
 
-```swift
-import OpenMedKit
+text = "<your clinical note here>"
+result = extract_pii(
+    text,
+    model_name="./OpenMed-PII-mLiteClinical-Base-135M-v1-mlx",
+    config=OpenMedConfig(backend="mlx"),
+    use_smart_merging=True,
+)
+
+print(result.entities)
 ```
 
-Then load your bundled CoreML model and label map:
+## Swift Quick Start with OpenMedKit
+
+OpenMedKit can download and run this MLX repo directly.
 
 ```swift
-import Foundation
 import OpenMedKit
 
-let modelFolder = Bundle.main.resourceURL!
-let modelURL = modelFolder.appendingPathComponent("OpenMedPII.mlmodelc")
-let labelsURL = modelFolder.appendingPathComponent("id2label.json")
+let modelDirectory = try await OpenMedModelStore.downloadMLXModel(
+    repoID: "OpenMed/OpenMed-PII-mLiteClinical-Base-135M-v1-mlx",
+    authToken: "<hugging-face-token-while-private>"
+)
 
 let openmed = try OpenMed(
-    modelURL: modelURL,
-    id2labelURL: labelsURL
+    backend: .mlx(modelDirectoryURL: modelDirectory)
+)
+
+let entities = try openmed.extractPII(
+    "<your clinical note here>"
 )
 ```
 
-This private MLX artifact remains the right choice for:
+Notes:
+
+- Swift MLX targets Apple Silicon macOS and physical iPhone/iPad hardware.
+- iOS Simulator is not a Swift MLX target.
+- If you already have a bundled `.mlmodelc` or `.mlpackage`, use the CoreML backend in OpenMedKit instead.
+
+## Artifact Notes
 
-- Python services on Apple Silicon
-- local MLX inference on macOS
-- private preconverted packaging on the Hub
+This repo uses the current legacy-compatible MLX layout:
 
-If a given architecture or environment cannot be exported cleanly as `safetensors`, OpenMed falls back to `weights.npz` so the model remains usable.
+- `config.json`
+- `id2label.json`
+- MLX weight files (`weights.safetensors` and/or `weights.npz`)
 
-## Credits
+Tokenizer assets are not bundled in this repo layout. OpenMed and OpenMedKit keep backward compatibility by falling back to the source tokenizer reference in `config.json` when needed.
 
-- Base checkpoint: [`OpenMed/OpenMed-PII-mLiteClinical-Base-135M-v1`](https://huggingface.co/OpenMed/OpenMed-PII-mLiteClinical-Base-135M-v1)
+## Links
+
+- Source checkpoint: [`OpenMed/OpenMed-PII-mLiteClinical-Base-135M-v1`](https://huggingface.co/OpenMed/OpenMed-PII-mLiteClinical-Base-135M-v1)
 - OpenMed GitHub: [https://github.com/maziyarpanahi/openmed](https://github.com/maziyarpanahi/openmed)
-- OpenMed website: [https://openmed.life](https://openmed.life)
-- MLX packaging and runtime support: [**OpenMed**](https://github.com/maziyarpanahi/openmed)
-- Swift runtime for Apple apps: **OpenMedKit** from the OpenMed repository
+- MLX backend docs: [https://openmed.life/docs/mlx-backend/](https://openmed.life/docs/mlx-backend/)
+- OpenMedKit docs: [https://openmed.life/docs/swift-openmedkit/](https://openmed.life/docs/swift-openmedkit/)