Update README.md

README.md

@@ -4,20 +4,16 @@ tags:
 - neuroimaging
 - fMRI
 library_name: transformers
-pipeline_tag: feature-extraction
-inference: false # This removes the "Hosted Inference" text box that doesn't work for fMRI
+inference: false
 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)
+    # Input shape: [batch, 1, 128]
     x = torch.randn(1, 1, 128)
-    outputs = model(x)
-    print(outputs["h"].shape) # [1, 128]
+    output = model(x)
 ---
 
 # NeuroCLR