IlyaGusev
/

mbart_ru_sum_gazeta

text2text-generation

Inference Endpoints

Model card Files Files and versions Community

IlyaGusev commited on Nov 23, 2021

Commit

1f3d3fd

•

1 Parent(s): b10b537

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -42,6 +42,7 @@ tokenizer = MBartTokenizer.from_pretrained(model_name)
 model = MBartForConditionalGeneration.from_pretrained(model_name)
 article_text = "..."
 input_ids = tokenizer(
     [article_text],
     max_length=600,
@@ -52,7 +53,7 @@ input_ids = tokenizer(
 output_ids = model.generate(
     input_ids=input_ids,
-    repetition_penalty=2.0
 )[0]
 summary = tokenizer.decode(output_ids, skip_special_tokens=True)

 model = MBartForConditionalGeneration.from_pretrained(model_name)
 article_text = "..."
 input_ids = tokenizer(
     [article_text],
     max_length=600,
 output_ids = model.generate(
     input_ids=input_ids,
+    repetition_penalty=3.0
 )[0]
 summary = tokenizer.decode(output_ids, skip_special_tokens=True)