huggingface_tools

HuggingFace 工具

Huggingface 工具 支持文本输入输出，可以直接使用 load_huggingface_tool 函数加载。

# 需要 transformers>=4.29.0 和 huggingface_hub>=0.14.1
pip install --upgrade transformers huggingface_hub > /dev/null

from langchain.agents import load_huggingface_tool

tool = load_huggingface_tool("lysandre/hf-model-downloads")

print(f"{tool.name}: {tool.description}")

    model_download_counter: 这是一个工具，它返回 Hugging Face Hub 上给定任务的最多下载模型。它接受类别名称（如文本分类、深度估计等），并返回检查点的名称

tool.run("text-classification")

    'facebook/bart-large-mnli'