Update README.md (#8544)

Modified Model in Action section. The class `AutoModelWithLMHead` is deprecated so changed it to `AutoModelForSeq2SeqLM` for encoder-decoder models. Removed duplicate eos token.
huggingface · Nov 18, 2020 · 26dc659 · 26dc659
1 parent 6c8fad4
commit 26dc659
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/model_cards/mrm8488/t5-base-finetuned-squadv2/README.md b/model_cards/mrm8488/t5-base-finetuned-squadv2/README.md
@@ -51,13 +51,13 @@ The training script is a slightly modified version of [this one](https://colab.r
 ## Model in Action 🚀
 
 ```python
-from transformers import AutoModelWithLMHead, AutoTokenizer
+from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 
 tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-squadv2")
-model = AutoModelWithLMHead.from_pretrained("mrm8488/t5-base-finetuned-squadv2")
+model = AutoModelForSeq2SeqLM.from_pretrained("mrm8488/t5-base-finetuned-squadv2")
 
 def get_answer(question, context):
-  input_text = "question: %s  context: %s </s>" % (question, context)
+  input_text = "question: %s  context: %s" % (question, context)
   features = tokenizer([input_text], return_tensors='pt')
 
   output = model.generate(input_ids=features['input_ids'],