Spaces:
Sleeping
Sleeping
File size: 11,433 Bytes
941cba6 9bf81ad 941cba6 3391dd2 941cba6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 |
# 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")
## 支持任务
| 域 | 任务 | 支持 | 文件 |
| :--------: | :---------------------------------------------------------------------------: | :---: | ------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- |
| 音频 | [音频分类](https://huggingface.co/tasks/audio-classification) | ✅ | [audio_classification（）](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/inference_client#huggingface_hub.InferenceClient.audio_classification) |
| 音频 | [音频到音频](https://huggingface.co/tasks/audio-to-audio) | ✅ | [audio_to_audio（）](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/inference_client#huggingface_hub.InferenceClient.audio_to_audio) |
| | [自动语音识别](https://huggingface.co/tasks/automatic-speech-recognition) | ✅ | [自动语音识别](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/inference_client#huggingface_hub.InferenceClient.automatic_speech_recognition) |
| | [Text-to-Speech](https://huggingface.co/tasks/text-to-speech) | ✅ | [text_to_speech（）](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/inference_client#huggingface_hub.InferenceClient.text_to_speech) |
| 计算机视觉 | [图像分类](https://huggingface.co/tasks/image-classification) | ✅ | [image_classification（）](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/inference_client#huggingface_hub.InferenceClient.image_classification) |
| | [图像分割](https://huggingface.co/tasks/image-segmentation) | ✅ | [image_segmentation（）](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/inference_client#huggingface_hub.InferenceClient.image_segmentation) |
| | [Image-to-Image](https://huggingface.co/tasks/image-to-image) | ✅ | [image_to_image（）](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/inference_client#huggingface_hub.InferenceClient.image_to_image) |
| | [图像到文本](https://huggingface.co/tasks/image-to-text) | ✅ | [image_to_text（）](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/inference_client#huggingface_hub.InferenceClient.image_to_text) |
| | [对象检测](https://huggingface.co/tasks/object-detection) | ✅ | [object_detection（）](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/inference_client#huggingface_hub.InferenceClient.object_detection) |
| | [文字转影像](https://huggingface.co/tasks/text-to-image) | ✅ | [text_to_image（）](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/inference_client#huggingface_hub.InferenceClient.text_to_image) |
| | [零拍摄图像分类](https://huggingface.co/tasks/zero-shot-image-classification) | ✅ | [zero_shot_image_classification（）](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/inference_client#huggingface_hub.InferenceClient.zero_shot_image_classification) |
| 多式 | [文档问题存档](https://huggingface.co/tasks/document-question-answering) | ✅ | [document_question_answering（）](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/inference_client#huggingface_hub.InferenceClient.document_question_answering) |
| | [视觉问题回答](https://huggingface.co/tasks/visual-question-answering) | ✅ | [visual_question_answering（）](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/inference_client#huggingface_hub.InferenceClient.visual_question_answering) |
| NLP | 会话 | | 已弃用，请使用聊天完成 |
| | [聊天完成](https://huggingface.co/tasks/text-generation) | ✅ | [chat_completion（）](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/inference_client#huggingface_hub.InferenceClient.chat_completion) |
| | [特征提取](https://huggingface.co/tasks/feature-extraction) | ✅ | [feature_extraction（）](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/inference_client#huggingface_hub.InferenceClient.feature_extraction) |
| | [填充掩膜](https://huggingface.co/tasks/fill-mask) | ✅ | [fill_mask（）](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/inference_client#huggingface_hub.InferenceClient.fill_mask) |
| | [问答](https://huggingface.co/tasks/question-answering) | ✅ | [question_answering（）](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/inference_client#huggingface_hub.InferenceClient.question_answering) |
| | [句子相似度](https://huggingface.co/tasks/sentence-similarity) | ✅ | [sentence_similarity（）](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/inference_client#huggingface_hub.InferenceClient.sentence_similarity) |
| | [总结](https://huggingface.co/tasks/summarization) | ✅ | [summarizing（）](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/inference_client#huggingface_hub.InferenceClient.summarization) |
| | [表格问题分类](https://huggingface.co/tasks/table-question-answering) | ✅ | [table_question_answering（）](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/inference_client#huggingface_hub.InferenceClient.table_question_answering) |
| | [文本分类](https://huggingface.co/tasks/text-classification) | ✅ | [text_classification（）](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/inference_client#huggingface_hub.InferenceClient.text_classification) |
| | [Generation Text一代](https://huggingface.co/tasks/text-generation) | ✅ | [text_generation（）](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/inference_client#huggingface_hub.InferenceClient.text_generation) |
| | [记号分类](https://huggingface.co/tasks/token-classification) | ✅ | [token_classification（）](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/inference_client#huggingface_hub.InferenceClient.token_classification) |
| | [翻译](https://huggingface.co/tasks/translation) | ✅ | [translation（）](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/inference_client#huggingface_hub.InferenceClient.translation) |
| | [零炮分类](https://huggingface.co/tasks/zero-shot-classification) | ✅ | [零炮分类](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/inference_client#huggingface_hub.InferenceClient.zero_shot_classification) |
| 表格 | [表格分类](https://huggingface.co/tasks/tabular-classification) | ✅ | [表分类](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/inference_client#huggingface_hub.InferenceClient.tabular_classification) |
| | [表格回归](https://huggingface.co/tasks/tabular-regression) | ✅ | [table_regression（）](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/inference_client#huggingface_hub.InferenceClient.tabular_regression) |
任务页面： https://huggingface.co/tasks |