Update README.md
Browse files
README.md
CHANGED
|
@@ -12,18 +12,11 @@ tags:
|
|
| 12 |
- pytorch
|
| 13 |
library_name: transformers
|
| 14 |
pipeline_tag: feature-extraction
|
| 15 |
-
#
|
| 16 |
-
model_creator: SaeedLab
|
| 17 |
-
model_type: custom
|
| 18 |
snippets:
|
| 19 |
- torch: |
|
| 20 |
from transformers import AutoModel
|
| 21 |
-
import torch
|
| 22 |
-
|
| 23 |
model = AutoModel.from_pretrained("SaeedLab/NeuroCLR", trust_remote_code=True)
|
| 24 |
-
# Input shape: [batch, 1, 128]
|
| 25 |
-
x = torch.randn(1, 1, 128)
|
| 26 |
-
output = model(x)
|
| 27 |
---
|
| 28 |
|
| 29 |
# NeuroCLR
|
|
@@ -139,5 +132,4 @@ The paper is under review. As soon as it is accepted, we will update this sectio
|
|
| 139 |
|
| 140 |
## Contact
|
| 141 |
|
| 142 |
-
For any additional questions or comments, contact Fahad Saeed (fsaeed@fiu.edu).
|
| 143 |
-
|
|
|
|
| 12 |
- pytorch
|
| 13 |
library_name: transformers
|
| 14 |
pipeline_tag: feature-extraction
|
| 15 |
+
inference: false # Disables the useless text-input box on the right
|
|
|
|
|
|
|
| 16 |
snippets:
|
| 17 |
- torch: |
|
| 18 |
from transformers import AutoModel
|
|
|
|
|
|
|
| 19 |
model = AutoModel.from_pretrained("SaeedLab/NeuroCLR", trust_remote_code=True)
|
|
|
|
|
|
|
|
|
|
| 20 |
---
|
| 21 |
|
| 22 |
# NeuroCLR
|
|
|
|
| 132 |
|
| 133 |
## Contact
|
| 134 |
|
| 135 |
+
For any additional questions or comments, contact Fahad Saeed (fsaeed@fiu.edu).
|
|
|