bkershner commited on
Commit
796f069
1 Parent(s): 7960fd0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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("paragon-analytics/ADRv1")
18
- model = AutoModelForSequenceClassification.from_pretrained("paragon-analytics/ADRv1")
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)