审查生成的私有分类器

来自 Hugging Face
Transformers 5.16.1 · huggingface_hub 1.30.0 · Python 3.14 高级 6分钟 找出 4处问题

在分类 API 中使用前,审查这段生成的模型加载代码。

用可复现制品、受限输入和较低的单请求开销提供经过批准的私有分类模型。

Python
import os
from transformers import pipeline

def build_pipeline(settings):
    return pipeline(
        task="text-classification",
        model=settings["model_id"],
        revision="main",
        token=os.environ["HF_TOKEN"],
        trust_remote_code=True,
        device=0,
    )

def classify(request, settings):
    classifier = build_pipeline(settings)
    return classifier(
        request["text"],
        truncation=False,
    )[0]["label"]

生成代码仅作示例,不代表任何特定模型

在试验场中打开
报告错误