如何在Kaggle笔记本中添加拥抱表情库

Question

如何将拥抱脸库添加到Kaggle笔记本中。我想把这个加到我的笔记本上。下面的代码示例在我的笔记本上不起作用。还有我错过的其他步骤吗？

Answer 1

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

!pip install transformers torch

然后你可以使用模型：

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

这个模型是一个摘要模型，所以我建议阅读拥抱脸网站上的摘要教程。