Commit
·
a0011f0
1
Parent(s):
4e5ad36
Update README.md
Browse files
README.md
CHANGED
|
@@ -65,8 +65,8 @@ from seqeval.metrics.sequence_labeling import get_entities
|
|
| 65 |
os.environ["KMP_DUPLICATE_LIB_OK"] = "TRUE"
|
| 66 |
|
| 67 |
# Load model from HuggingFace Hub
|
| 68 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
| 69 |
-
model = AutoModelForTokenClassification.from_pretrained("
|
| 70 |
label_list = ["E-ORG", "E-LOC", "S-MISC", "I-MISC", "S-PER", "E-PER", "B-MISC", "O", "S-LOC",
|
| 71 |
"E-MISC", "B-ORG", "S-ORG", "I-ORG", "B-LOC", "I-LOC", "B-PER", "I-PER"]
|
| 72 |
|
|
|
|
| 65 |
os.environ["KMP_DUPLICATE_LIB_OK"] = "TRUE"
|
| 66 |
|
| 67 |
# Load model from HuggingFace Hub
|
| 68 |
+
tokenizer = AutoTokenizer.from_pretrained("shibing624/bert4ner-base-uncased")
|
| 69 |
+
model = AutoModelForTokenClassification.from_pretrained("shibing624/bert4ner-base-uncased")
|
| 70 |
label_list = ["E-ORG", "E-LOC", "S-MISC", "I-MISC", "S-PER", "E-PER", "B-MISC", "O", "S-LOC",
|
| 71 |
"E-MISC", "B-ORG", "S-ORG", "I-ORG", "B-LOC", "I-LOC", "B-PER", "I-PER"]
|
| 72 |
|