README / README.md
rvignav's picture
Update README.md
ebe1a11
|
raw
history blame
463 Bytes
metadata
title: README
emoji: πŸš€
colorFrom: red
colorTo: purple
sdk: static
pinned: false

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/biobert-finetuned-prior-rmv"
tokenizer = AutoTokenizer.from_pretrained(modelname)
model = AutoModelForTokenClassification.from_pretrained(modelname)