支持类型.md

hf的功能

文字

https://huggingface.co/docs/huggingface_hub/main/en/guides/inference#run-inference-on-servers

from huggingface_hub import InferenceClient messages = [{"role": "user", "content": "中国的首都是哪里?"}] client = InferenceClient("meta-llama/Llama-3.2-11B-Vision-Instruct") client.chat_completion(messages, max_tokens=100)

from huggingface_hub import InferenceClient messages = [{"role": "user", "content": "中国的首都是哪里?"}] client = InferenceClient("google/gemma-2-2b-it") client.chat_completion(messages, max_tokens=100)

图片

中文识别不好

from huggingface_hub import InferenceClient client = InferenceClient() image = client.text_to_image("An astronaut riding a horse on the moon.") image.save("/Users/sanbo/Desktop/astronaut.png") # 'image' is a PIL.Image object

from huggingface_hub import InferenceClient client = InferenceClient("black-forest-labs/FLUX.1-dev") image = client.text_to_image("一个天上飞的乌龟.") image.save("/Users/sanbo/Desktop/astronaut.png") # 'image' is a PIL.Image object

视觉问答visual_question_answering

from huggingface_hub import InferenceClient client = InferenceClient() client.visual_question_answering( image="https://huggingface.co/datasets/mishig/sample_images/resolve/main/tiger.jpg", question="What is the animal doing?" )

翻译

from huggingface_hub import InferenceClient client = InferenceClient() client.translation("My name is Wolfgang and I live in Berlin") 'Mein Name ist Wolfgang und ich lebe in Berlin.' client.translation("My name is Wolfgang and I live in Berlin", model="Helsinki-NLP/opus-mt-en-zh")

Helsinki-NLP/opus-mt-zh-en

使用特定模型

client = InferenceClient(model="prompthero/openjourney-v4") client.text_to_image("xxx")

方式二 client = InferenceClient() client.text_to_image(..., model="prompthero/openjourney-v4")

客户端请求

from huggingface_hub import InferenceClient client = InferenceClient() response = client.post(json={"inputs": "An astronaut riding a horse on the moon."}, model="stabilityai/stable-diffusion-2-1") response.content

支持模型

https://huggingface.co/models?other=conversational&sort=likes

client = InferenceClient("meta-llama/Meta-Llama-3-8B-Instruct") client = InferenceClient("Qwen/Qwen2.5-Coder-32B-Instruct")

支持任务

域	任务	支持	文件
音频	音频分类	✅	audio_classification（）
音频	音频到音频	✅	audio_to_audio（）
	自动语音识别	✅	自动语音识别
	Text-to-Speech	✅	text_to_speech（）
计算机视觉	图像分类	✅	image_classification（）
	图像分割	✅	image_segmentation（）
	Image-to-Image	✅	image_to_image（）
	图像到文本	✅	image_to_text（）
	对象检测	✅	object_detection（）
	文字转影像	✅	text_to_image（）
	零拍摄图像分类	✅	zero_shot_image_classification（）
多式	文档问题存档	✅	document_question_answering（）
	视觉问题回答	✅	visual_question_answering（）
NLP	会话		已弃用，请使用聊天完成
	聊天完成	✅	chat_completion（）
	特征提取	✅	feature_extraction（）
	填充掩膜	✅	fill_mask（）
	问答	✅	question_answering（）
	句子相似度	✅	sentence_similarity（）
	总结	✅	summarizing（）
	表格问题分类	✅	table_question_answering（）
	文本分类	✅	text_classification（）
	Generation Text一代	✅	text_generation（）
	记号分类	✅	token_classification（）
	翻译	✅	translation（）
	零炮分类	✅	零炮分类
表格	表格分类	✅	表分类
	表格回归	✅	table_regression（）

任务页面： https://huggingface.co/tasks