MRNH
/

mbart-italian-grammar-corrector

Text2Text Generation

grammatical error correction

Inference Endpoints

Model card Files Files and versions Community

matteo commited on Jun 28, 2023

Commit

2e9d7dd

·

1 Parent(s): 5aedc32

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -7,14 +7,14 @@ pipeline_tag: translation
 To initialize the model:
     ```python
-    from transformers import MBartForConditionalGeneration
     model = MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50", output_hidden_states=True)
     ```
 To generate text using the model:
     ```python
     input_ids = tokenizer.encode("Input text", return_tensors="pt")
     output = model.generate(input_ids)
     ```

 To initialize the model:
     ```python
+    from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
     model = MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50", output_hidden_states=True)
     ```
 To generate text using the model:
     ```python
+    tokenizer = MBart50TokenizerFast.from_pretrained("facebook/mbart-large-50", src_lang="it_IT", tgt_lang="it_IT")
     input_ids = tokenizer.encode("Input text", return_tensors="pt")
     output = model.generate(input_ids)
     ```