kalkiek95 commited on
Commit
09d29f8
·
verified ·
1 Parent(s): 996ff9d
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -21,7 +21,7 @@ import torch
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
 
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