language:

• 希伯来语 pipeline_tag: 文本摘要

from transformers import AutoModelForSeq2SeqLM, AutoTokenizer, SummarizationPipeline


model_name = "imvladikon/het5_small_summarization"
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)
summarizer = SummarizationPipeline(model=model, tokenizer=tokenizer)

示例

text = """
法国持续动荡：在暴徒与安全部队在全国范围内发生四天暴力冲突后，周六举行了17岁阿尔及利亚少年纳赫尔（Nahel）的葬礼，他因涉嫌偷车被警察开枪打死。应其家人要求，葬礼以仅限亲友参加的小规模仪式进行。当纳赫尔的灵柩从南泰尔市清真寺运出时，数千人高呼"为他讨回公道"。与此同时，法国警方正为全国数十个骚乱点持续部署，夜间逮捕了1300人。法国内政部宣布，骚乱期间有1350辆汽车被焚毁，234栋建筑遭纵火。数据显示，还有200个购物中心、200家超市和250家银行网点受损。
""".strip()
summarizer(text,
           max_length=50,
           num_beams=4,
           no_repeat_ngram_size=2,
           early_stopping=True)[0]["summary_text"]
#17岁阿尔及利亚少年因涉嫌偷车被警察击毙。其葬礼以仅限亲友参加的小规模仪式举行