Views
No views yet
| Label | Description |
|---|---|
B-problem / I-problem | Clinical problem or diagnosis |
B-treatment / I-treatment | Treatment or medication |
B-test / I-test | Clinical test or procedure |
O | Outside any entity |
1from transformers import AutoTokenizer
2from optimum.onnxruntime import ORTModelForTokenClassification
3
4tokenizer = AutoTokenizer.from_pretrained("sidupadhyay/bert-base-uncased_clinical-ner-ONNX")
5model = ORTModelForTokenClassification.from_pretrained("sidupadhyay/bert-base-uncased_clinical-ner-ONNX")
6
7inputs = tokenizer("The patient was prescribed aspirin for chest pain.", return_tensors="pt")
8outputs = model(**inputs)