Update README.md

README.md

@@ -4,20 +4,23 @@ tags:
 - neuroimaging
 - fMRI
 - pytorch
-
 library_name: transformers
 inference: false
-snippets:
-- torch: |
-    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)
+model-index:
+- name: NeuroCLR
+  results: []
+  usage:
+    torch: |
+      from transformers import AutoModel
+      import torch
+
+      # Load NeuroCLR SSL Encoder
+      model = AutoModel.from_pretrained("SaeedLab/NeuroCLR", trust_remote_code=True)
+      
+      # Input: [batch, 1, 128] (raw fMRI time series)
+      x = torch.randn(1, 1, 128)
+      outputs = model(x)
+      print(outputs["h"].shape)
 ---
 
 # NeuroCLR