gilbert / README.md
rvignav's picture
Update README.md
1f2165b
metadata
language:
  - py
metrics:
  - f1

To use our fine-tuned BioBERT model to remove references to priors from radiology reports, run the following:

from transformers import AutoTokenizer, AutoModelForTokenClassification

modelname = "rajpurkarlab/gilbert"
tokenizer = AutoTokenizer.from_pretrained(modelname)
model = AutoModelForTokenClassification.from_pretrained(modelname)