philippelaban
commited on
Commit
•
f2b343d
1
Parent(s):
c83d0cb
Update README.md
Browse files
README.md
CHANGED
@@ -22,8 +22,8 @@ Example usage of the model:
|
|
22 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer
|
23 |
import itertools
|
24 |
|
25 |
-
ae_tokenizer = AutoTokenizer.from_pretrained("/
|
26 |
-
ae_model = AutoModelForSequenceClassification.from_pretrained("/
|
27 |
|
28 |
question = "When will the recession happen?"
|
29 |
answers = ["probably next January", "never", "we're already in a recession", "it won't happen", "it's going on right now", "not before next year", "upcoming January-March"]
|
|
|
22 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer
|
23 |
import itertools
|
24 |
|
25 |
+
ae_tokenizer = AutoTokenizer.from_pretrained("Salesforce/qa_consolidation")
|
26 |
+
ae_model = AutoModelForSequenceClassification.from_pretrained("Salesforce/qa_consolidation").eval()
|
27 |
|
28 |
question = "When will the recession happen?"
|
29 |
answers = ["probably next January", "never", "we're already in a recession", "it won't happen", "it's going on right now", "not before next year", "upcoming January-March"]
|