lincoln
/

mbart-mlsum-automatic-summarization

text2text-generation

Model card Files Files and versions

ahm commited on May 3, 2021

Commit

53d02bf

·

1 Parent(s): 6f4e37f

fix exemple

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -28,8 +28,8 @@ from transformers import SummarizationPipeline
 model_name = 'lincoln/mbart-mlsum-automatic-summarization'
-loaded_tokenizer = AutoTokenizer.from_pretrained(model_name)
-loaded_model = AutoModelForSequenceClassification.from_pretrained(model_name)
 nlp = SummarizationPipeline(model=loaded_model, tokenizer=loaded_tokenizer)
 nlp("""

 model_name = 'lincoln/mbart-mlsum-automatic-summarization'
+loaded_tokenizer = MBartTokenizer.from_pretrained(model_name)
+loaded_model = MBartForConditionalGeneration.from_pretrained(model_name)
 nlp = SummarizationPipeline(model=loaded_model, tokenizer=loaded_tokenizer)
 nlp("""