Update README.md

README.md

@@ -10,7 +10,20 @@ tags:
 - domain-generalization
 - transfer-learning
 - pytorch
-# library_name: transformers
+library_name: transformers
+pipeline_tag: feature-extraction
+# This section overrides the default "Use in Transformers" popup text
+model_creator: SaeedLab
+model_type: custom
+snippets:
+- torch: |
+    from transformers import AutoModel
+    import torch
+
+    model = AutoModel.from_pretrained("SaeedLab/NeuroCLR", trust_remote_code=True)
+    # Input shape: [batch, 1, 128]
+    x = torch.randn(1, 1, 128)
+    output = model(x)
 ---
 
 # NeuroCLR