shibing624
commited on
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 |
|