Update README.md

README.md

@@ -3,6 +3,8 @@ tags:
 - self-supervised-learning
 - neuroimaging
 - fMRI
+- pytorch
+
 library_name: transformers
 inference: false
 snippets:
@@ -10,7 +12,12 @@ snippets:
     from transformers import AutoModel
     import torch
 
+    # Load NeuroCLR directly (the pipeline block should now be gone)
     model = AutoModel.from_pretrained("SaeedLab/NeuroCLR", trust_remote_code=True)
+    
+    # Input shape: [batch, 1, 128]
+    x = torch.randn(1, 1, 128)
+    outputs = model(x)
 ---
 
 # NeuroCLR