philippelaban commited on
Commit
f2b343d
1 Parent(s): c83d0cb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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("/export/share/plaban/models/qa_consolidation/")
26
- ae_model = AutoModelForSequenceClassification.from_pretrained("/export/share/plaban/models/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"]
 
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"]