Update README.md

README.md

@@ -1,22 +1,23 @@
 ---
 tags:
 - self-supervised-learning
-- contrastive-learning
 - neuroimaging
 - fMRI
-- rs-fMRI
-- neuroscience
-- representation-learning
-- domain-generalization
-- transfer-learning
-- pytorch
 library_name: transformers
 pipeline_tag: feature-extraction
-inference: false  # Disables the useless text-input box on the right
+inference: false # This removes the "Hosted Inference" text box that doesn't work for fMRI
 snippets:
 - torch: |
     from transformers import AutoModel
+    import torch
+
+    # Load the 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) # [1, 128]
 ---
 
 # NeuroCLR