Update README.md

README.md

@@ -20,11 +20,12 @@ The training script can be found [here](https://github.com/McGill-NLP/MLQuestion
 ## Model in Action 🚀
 
 ```python
-from transformers import AutoModel, BartTokenizer
+
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 #Load the tokenizer
-tokenizer = BartTokenizer.from_pretrained('facebook/bart-base')
+tokenizer = AutoTokenizer.from_pretrained("geekydevu/bart-qg-mlquestions-backtraining")
 #Load the model
-model = AutoModelForSeq2SeqLM.from_pretrained("McGill-NLP/bart-qg-nq-checkpoint")
+model = AutoModelForSeq2SeqLM.from_pretrained("geekydevu/bart-qg-mlquestions-backtraining")
 ```
 
 ## Citation