python - key 错误 : "Unknown task summarization, available tasks are [' feature-extraction', 'sentiment-analysis' 、 'ner' 、 'question-answering' 、 'fill-mask' ]"

Question

我正在尝试对该数据框中的一列进行文本摘要。我按照本网站( https://towardsdatascience.com/summarization-of-covid-research-papers-using-bart-model-5b109a6669a6 )中提到的步骤进行操作，但出现此错误
KeyError:“未知任务摘要，可用任务是 ['特征提取'、'情感分析'、'ner'、'问题回答'、'填充掩码']”
Screenshot of Error

Answer 1

你应该更新你的变压器库:

pip install -U transformers

from transformers import pipeline

# use bart in pytorch
summarizer = pipeline("summarization")
summarizer("An apple a day, keeps the doctor away", min_length=5, max_length=20)

# use t5 in tf
summarizer = pipeline("summarization", model="t5-base", tokenizer="t5-base", framework="tf")
summarizer("An apple a day, keeps the doctor away", min_length=5, max_length=20)