在分类 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"]
生成代码仅作示例,不代表任何特定模型