Update README.md
Browse files
README.md
CHANGED
@@ -19,14 +19,16 @@ To initialize the model:
|
|
19 |
model = MBartForConditionalGeneration.from_pretrained("MRNH/mbart-english-grammar-corrector")
|
20 |
|
21 |
|
22 |
-
|
23 |
|
24 |
|
25 |
tokenizer = MBart50TokenizerFast.from_pretrained("MRNH/mbart-english-grammar-corrector", src_lang="en_XX", tgt_lang="en_XX")
|
26 |
|
27 |
input = tokenizer("I was here yesterday to studying",
|
28 |
text_target="I was here yesterday to study", return_tensors='pt')
|
29 |
-
|
|
|
|
|
30 |
output = model.generate(input["input_ids"],attention_mask=input["attention_mask"],
|
31 |
forced_bos_token_id=tokenizer_it.lang_code_to_id["en_XX"])
|
32 |
|
|
|
19 |
model = MBartForConditionalGeneration.from_pretrained("MRNH/mbart-english-grammar-corrector")
|
20 |
|
21 |
|
22 |
+
Use the tokenizer:
|
23 |
|
24 |
|
25 |
tokenizer = MBart50TokenizerFast.from_pretrained("MRNH/mbart-english-grammar-corrector", src_lang="en_XX", tgt_lang="en_XX")
|
26 |
|
27 |
input = tokenizer("I was here yesterday to studying",
|
28 |
text_target="I was here yesterday to study", return_tensors='pt')
|
29 |
+
|
30 |
+
To generate text using the model:
|
31 |
+
|
32 |
output = model.generate(input["input_ids"],attention_mask=input["attention_mask"],
|
33 |
forced_bos_token_id=tokenizer_it.lang_code_to_id["en_XX"])
|
34 |
|