sociocom
/

MedTXTNER

Token Classification

Model card Files Files and versions

Tomohiro commited on Apr 22, 2025

Commit

59c0418

·

verified ·

1 Parent(s): 446ae99

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -34,7 +34,7 @@ from transformers import AutoTokenizer, AutoModelForTokenClassification
 model_dir = "sociocom/MedTXTNER"
 model     = AutoModelForTokenClassification.from_pretrained(model_dir)
-tokenizer = AutoTokenizer.from_pretrained(checkpoint_dir, use_fast=True)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)
 model.eval()

 model_dir = "sociocom/MedTXTNER"
 model     = AutoModelForTokenClassification.from_pretrained(model_dir)
+tokenizer = AutoTokenizer.from_pretrained(model_dir, use_fast=True)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)
 model.eval()