cbdb commited on
Commit
e6d0f20
1 Parent(s): 9cfa1f1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -28,7 +28,7 @@ Here is how to use this model to get the features of a given text in PyTorch:
28
  ```python
29
  from transformers import AutoTokenizer, AutoModelForTokenClassification
30
 
31
- PRETRAINED = "cbdb/NameSplitter"
32
  tokenizer = AutoTokenizer.from_pretrained(PRETRAINED)
33
  model = AutoModelForTokenClassification.from_pretrained(PRETRAINED)
34
  ```
 
28
  ```python
29
  from transformers import AutoTokenizer, AutoModelForTokenClassification
30
 
31
+ PRETRAINED = "cbdb/PersonAndBookTitleSplitter"
32
  tokenizer = AutoTokenizer.from_pretrained(PRETRAINED)
33
  model = AutoModelForTokenClassification.from_pretrained(PRETRAINED)
34
  ```