jpwahle commited on
Commit
0763962
1 Parent(s): 154c202

update model card

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -27,10 +27,10 @@ Additional information about this model:
27
  The model can be loaded to perform a few-shot classification like so:
28
 
29
  ```py
30
- from transformers import AutoModelForConditionalGeneration, AutoTokenizer
31
 
32
- AutoModelForConditionalGeneration.from_pretrained("jpelhaw/t5-word-sense-disambiguation")
33
- AutoTokenizer("jpelhaw/t5-word-sense-disambiguation")
34
 
35
  input = 'question: which description describes the word " peculiarities " best in the following context? \
36
  descriptions: [ " an odd or unusual characteristic " , " a distinguishing trait " , or " something unusual -- perhaps worthy of collecting " ] \
 
27
  The model can be loaded to perform a few-shot classification like so:
28
 
29
  ```py
30
+ from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
31
 
32
+ AutoModelForSeq2SeqLM.from_pretrained("jpelhaw/t5-word-sense-disambiguation")
33
+ AutoTokenizer.from_pretrained("jpelhaw/t5-word-sense-disambiguation")
34
 
35
  input = 'question: which description describes the word " peculiarities " best in the following context? \
36
  descriptions: [ " an odd or unusual characteristic " , " a distinguishing trait " , or " something unusual -- perhaps worthy of collecting " ] \