Fix Typo
Browse files
README.md
CHANGED
@@ -21,7 +21,7 @@ import torch
|
|
21 |
import numpy as np
|
22 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
23 |
|
24 |
-
MODEL_NAME = "
|
25 |
CHUNK_SIZE = 512 # Neurobiber was trained with max_length=512
|
26 |
|
27 |
# List of the 96 features that Neurobiber can predict
|
|
|
21 |
import numpy as np
|
22 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
23 |
|
24 |
+
MODEL_NAME = "Blablablab/neurobiber"
|
25 |
CHUNK_SIZE = 512 # Neurobiber was trained with max_length=512
|
26 |
|
27 |
# List of the 96 features that Neurobiber can predict
|