Update README.md
Browse files
README.md
CHANGED
@@ -18,4 +18,4 @@ To generate text using the model:
|
|
18 |
|
19 |
tokenizer = MBart50TokenizerFast.from_pretrained("MRNH/finetuned-mbart-it-gec", src_lang="it_IT", tgt_lang="it_IT")
|
20 |
input = tokenizer("I was here yesterday to studying",text_target="I was here yesterday to study", return_tensors='pt')
|
21 |
-
output =
|
|
|
18 |
|
19 |
tokenizer = MBart50TokenizerFast.from_pretrained("MRNH/finetuned-mbart-it-gec", src_lang="it_IT", tgt_lang="it_IT")
|
20 |
input = tokenizer("I was here yesterday to studying",text_target="I was here yesterday to study", return_tensors='pt')
|
21 |
+
output = model.generate(input["input_ids"],attention_mask=input["attention_mask"],forced_bos_token_id=tokenizer_it.lang_code_to_id["it_IT"])
|