d4data
/

biomedical-ner-all

Token Classification

Token Classification

Model card Files Files and versions

dreji18 commited on Jun 19, 2022

Commit

74ffa29

·

1 Parent(s): 0545615

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -34,8 +34,8 @@ The easiest way is to load the inference api from huggingface and second method
 from transformers import pipeline
 from transformers import AutoTokenizer, AutoModelForTokenClassification
-tokenizer = AutoTokenizer.from_pretrained("d4data/biomedical-all")
-model = AutoModelForTokenClassification.from_pretrained("d4data/biomedical-all")
 pipe = pipeline("ner", model=model, tokenizer=tokenizer, aggregation_strategy="simple") # pass device=0 if using gpu
 pipe("""The patient reported no recurrence of palpitations at follow-up 6 months after the ablation.""")

 from transformers import pipeline
 from transformers import AutoTokenizer, AutoModelForTokenClassification
+tokenizer = AutoTokenizer.from_pretrained("d4data/biomedical-ner-all")
+model = AutoModelForTokenClassification.from_pretrained("d4data/biomedical-ner-all")
 pipe = pipeline("ner", model=model, tokenizer=tokenizer, aggregation_strategy="simple") # pass device=0 if using gpu
 pipe("""The patient reported no recurrence of palpitations at follow-up 6 months after the ablation.""")