roberthsu2003
/

models_for_ner

Token Classification

Generated from Trainer

Eval Results (legacy)

Model card Files Files and versions

roberthsu2003 commited on Mar 2, 2025

Commit

7d467f2

·

verified ·

1 Parent(s): 58ff653

Update README.md

Files changed (1) hide show

README.md +3 -1

README.md CHANGED Viewed

@@ -71,7 +71,9 @@ model = AutoModelForTokenClassification.from_pretrained('roberthsu2003/models_fo
 tokenizer = AutoTokenizer.from_pretrained('roberthsu2003/models_for_ner')
 # The label mapping (you might need to adjust this based on your training)
-label_list = ['O', 'B-PER', 'I-PER', 'B-ORG', 'I-ORG', 'B-LOC', 'I-LOC']
 def predict_ner(text):
     """Predicts NER tags for a given text using the loaded model."""

 tokenizer = AutoTokenizer.from_pretrained('roberthsu2003/models_for_ner')
 # The label mapping (you might need to adjust this based on your training)
+#['O', 'B-PER', 'I-PER', 'B-ORG', 'I-ORG', 'B-LOC', 'I-LOC']
+label_list = list(model.config.id2label.values())
 def predict_ner(text):
     """Predicts NER tags for a given text using the loaded model."""