IlyaGusev
/

mbart_ru_sum_gazeta

text2text-generation

Inference Endpoints

Model card Files Files and versions Community

IlyaGusev commited on Nov 23, 2021

Commit

a423aae

•

1 Parent(s): 465adf9

Update README.md

Files changed (1) hide show

README.md +1 -4

README.md CHANGED Viewed

@@ -41,10 +41,7 @@ input_ids = tokenizer.prepare_seq2seq_batch(
 output_ids = model.generate(
     input_ids=input_ids,
-    max_length=162,
-    no_repeat_ngram_size=3,
-    num_beams=5,
-    top_k=0
 )[0]
 summary = tokenizer.decode(output_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False)

 output_ids = model.generate(
     input_ids=input_ids,
+    no_repeat_ngram_size=3
 )[0]
 summary = tokenizer.decode(output_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False)