SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[Doctests] Correct task summary (#16644)

Browse Source
This commit is contained in:
Patrick von Platen
2022-04-11 14:59:35 +02:00
committed by GitHub
parent 6ef7186b5d
commit 098b002644
1 changed files with 7 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
docs/source/en/task_summary.mdx
View File

@@ -1090,16 +1090,15 @@ The following examples demonstrate how to use a [`pipeline`] and a model and tok
>>> from transformers import pipeline >>> from transformers import pipeline
>>> vision_classifier = pipeline(task="image-classification") >>> vision_classifier = pipeline(task="image-classification")
>>> vision_classifier( >>> result = vision_classifier(
... images="https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/pipeline-cat-chonk.jpeg" ... images="https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/pipeline-cat-chonk.jpeg"
... ) ... )
[{'label': 'lynx, catamount', 'score': 0.4403027892112732}, >>> print("\n".join([f"Class {d['label']} with score {round(d['score'], 4)}" for d in result]))
{'label': 'cougar, puma, catamount, mountain lion, painter, panther, Felis concolor', Class lynx, catamount with score 0.4335
'score': 0.03433405980467796}, Class cougar, puma, catamount, mountain lion, painter, panther, Felis concolor with score 0.0348
{'label': 'snow leopard, ounce, Panthera uncia', Class snow leopard, ounce, Panthera uncia with score 0.0324
'score': 0.032148055732250214}, Class Egyptian cat with score 0.0239
{'label': 'Egyptian cat', 'score': 0.02353910356760025}, Class tiger cat with score 0.0229
{'label': 'tiger cat', 'score': 0.023034192621707916}]
``` ```
The general process for using a model and feature extractor for image classification is: The general process for using a model and feature extractor for image classification is: