gator-base / index.py
zaeemzafar's picture
Create index.py
83014c4 verified
raw
history blame contribute delete
403 Bytes
from transformers import AutoModel, AutoTokenizer, AutoConfig
tokenizer= AutoTokenizer.from_pretrained('UFNLP/gatortron-base')
config=AutoConfig.from_pretrained('UFNLP/gatortron-base')
mymodel=AutoModel.from_pretrained('UFNLP/gatortron-base')
encoded_input=tokenizer("Bone scan: Negative for distant metastasis.", return_tensors="pt")
encoded_output = mymodel(**encoded_input)
print (encoded_output)