Update README.md
Browse files
README.md
CHANGED
@@ -14,8 +14,8 @@ from transformers import RobertaTokenizer, RobertaModel
|
|
14 |
from transformers import AutoModelForSequenceClassification
|
15 |
from transformers import TFAutoModelForSequenceClassification
|
16 |
from transformers import AutoTokenizer
|
17 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
18 |
-
model = AutoModelForSequenceClassification.from_pretrained("
|
19 |
def adr_predict(x):
|
20 |
encoded_input = tokenizer(x, return_tensors='pt')
|
21 |
output = model(**encoded_input)
|
|
|
14 |
from transformers import AutoModelForSequenceClassification
|
15 |
from transformers import TFAutoModelForSequenceClassification
|
16 |
from transformers import AutoTokenizer
|
17 |
+
tokenizer = AutoTokenizer.from_pretrained("UVA-MSBA/M4T8")
|
18 |
+
model = AutoModelForSequenceClassification.from_pretrained("UVA-MSBA/M4T8")
|
19 |
def adr_predict(x):
|
20 |
encoded_input = tokenizer(x, return_tensors='pt')
|
21 |
output = model(**encoded_input)
|