Update README.md

README.md

@@ -58,6 +58,29 @@ Both subfolders use **custom code** (Transformer encoder + projector; and a ResN
 
 ---
 
+## Usage (PyTorch)
+
+```python
+import torch
+from transformers import AutoModel
+
+model = AutoModel.from_pretrained(
+    "SaeedLab/NeuroCLR",
+    subfolder="pretraining",
+    trust_remote_code=True
+)
+
+model.eval()
+
+x = torch.randn(4, 1, 128)  # [batch, seq_len, feature_dim]
+
+with torch.no_grad():
+    outputs = model(x)
+
+print(outputs["h"].shape)
+print(outputs["z"].shape)
+```
+
 ## Model Usage
 
 ### Requirements