MRNH commited on
Commit
48d3361
1 Parent(s): d3c83a7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -10,7 +10,7 @@ To initialize the model:
10
 
11
 
12
  from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
13
- model = MBartForConditionalGeneration.from_pretrained("MRNH/finetuned-mbart-it-gec", output_hidden_states=True)
14
 
15
 
16
  To generate text using the model:
 
10
 
11
 
12
  from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
13
+ model = MBartForConditionalGeneration.from_pretrained("MRNH/finetuned-mbart-it-gec")
14
 
15
 
16
  To generate text using the model: