如何将拥抱面库添加到 Kaggle 笔记本

Question

如何将拥抱脸库添加到 Kaggle 笔记本中。我想将 this one 添加到我的笔记本中。下面的代码示例在我的笔记本中不起作用。我错过了一些额外的步骤吗？

Answer 1

huggingface 模型集线器的所有模型都可以在安装库后使用。这个特别是只在 PyTorch 中可用，所以你应该同时安装 transformers 和 torch:

!pip install transformers torch

然后您可以使用模型：

from transformers import BartForConditionalGeneration

model = BartForConditionalGeneration.from_pretrained("facebook/bart-large-cnn")

此模型是一个摘要模型，因此我建议阅读 Hugging Face 网站上的 summarization tutorial。