Update README.md
Browse files
README.md
CHANGED
@@ -20,11 +20,12 @@ The training script can be found [here](https://github.com/McGill-NLP/MLQuestion
|
|
20 |
## Model in Action 🚀
|
21 |
|
22 |
```python
|
23 |
-
|
|
|
24 |
#Load the tokenizer
|
25 |
-
tokenizer =
|
26 |
#Load the model
|
27 |
-
model = AutoModelForSeq2SeqLM.from_pretrained("
|
28 |
```
|
29 |
|
30 |
## Citation
|
|
|
20 |
## Model in Action 🚀
|
21 |
|
22 |
```python
|
23 |
+
|
24 |
+
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
25 |
#Load the tokenizer
|
26 |
+
tokenizer = AutoTokenizer.from_pretrained("geekydevu/bart-qg-mlquestions-backtraining")
|
27 |
#Load the model
|
28 |
+
model = AutoModelForSeq2SeqLM.from_pretrained("geekydevu/bart-qg-mlquestions-backtraining")
|
29 |
```
|
30 |
|
31 |
## Citation
|