Hugging Face's logo

optimum
/
vit-base-patch16-224

Image Classification
Transformers
ONNX
vit
vision
Inference Endpoints
Model card Files Community
2

Specify the head of the model in the model card

#2
by regisss HF staff - opened
base: refs/heads/main
←
from: refs/pr/2
Discussion Files changed
+16
-19
regisss
 Hugging Face Optimum org
•
edited Dec 14, 2022

It is not clear in the model card which task this model performs.

Specify the head of the model in the model cardf83e0163
regisss
 Hugging Face Optimum org

@Jingya Let me know if this update is good enough.

Jingya
 Hugging Face Optimum org

Hi @regisss ,

Yeah, that looks clearer to me!
Maybe we can replace the snippet from the transformers' one with the one using ORTModel? (but it is just an extra plus. Feel free to merge it.)

Update code snippetc9ee1c31
regisss
 Hugging Face Optimum org

@Jingya I modified the code snippet to add an example with Optimum pipelines.

Jingya
 Hugging Face Optimum org

It looks great!!!! Let's merge it :D

regisss changed pull request status to merged

Sign up or log in to comment