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

Question

如何将拥抱脸库添加到Kaggle笔记本中。我想在笔记本中添加this one。下面的代码示例不适用于我拥有的笔记本电脑。我还错过了其他一些步骤吗？

Answer 1

一旦安装了库，就可以使用表面模型中心的所有模型。特别是仅在PyTorch中可用，因此您应该同时安装transformers和torch：

!pip install transformers torch

然后您可以使用模型：

from transformers import BartForConditionalGeneration

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

此模型是一种汇总模型，因此，我建议阅读Hugging Face网站上的summarization tutorial。