Spaces:
No application file
No application file
Create index.py
Browse files
index.py
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from transformers import AutoModel, AutoTokenizer, AutoConfig
|
| 2 |
+
|
| 3 |
+
tokenizer= AutoTokenizer.from_pretrained('UFNLP/gatortron-base')
|
| 4 |
+
config=AutoConfig.from_pretrained('UFNLP/gatortron-base')
|
| 5 |
+
mymodel=AutoModel.from_pretrained('UFNLP/gatortron-base')
|
| 6 |
+
|
| 7 |
+
encoded_input=tokenizer("Bone scan: Negative for distant metastasis.", return_tensors="pt")
|
| 8 |
+
encoded_output = mymodel(**encoded_input)
|
| 9 |
+
print (encoded_output)
|