ahm
commited on
Commit
•
53d02bf
1
Parent(s):
6f4e37f
fix exemple
Browse files
README.md
CHANGED
@@ -28,8 +28,8 @@ from transformers import SummarizationPipeline
|
|
28 |
|
29 |
model_name = 'lincoln/mbart-mlsum-automatic-summarization'
|
30 |
|
31 |
-
loaded_tokenizer =
|
32 |
-
loaded_model =
|
33 |
|
34 |
nlp = SummarizationPipeline(model=loaded_model, tokenizer=loaded_tokenizer)
|
35 |
nlp("""
|
|
|
28 |
|
29 |
model_name = 'lincoln/mbart-mlsum-automatic-summarization'
|
30 |
|
31 |
+
loaded_tokenizer = MBartTokenizer.from_pretrained(model_name)
|
32 |
+
loaded_model = MBartForConditionalGeneration.from_pretrained(model_name)
|
33 |
|
34 |
nlp = SummarizationPipeline(model=loaded_model, tokenizer=loaded_tokenizer)
|
35 |
nlp("""
|