How to use optimum/vit-base-patch16-224 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="optimum/vit-base-patch16-224") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")
# Load model directly from transformers import AutoImageProcessor, AutoModelForImageClassification processor = AutoImageProcessor.from_pretrained("optimum/vit-base-patch16-224") model = AutoModelForImageClassification.from_pretrained("optimum/vit-base-patch16-224")
It is not clear in the model card which task this model performs.
@Jingya Let me know if this update is good enough.
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.)
@Jingya I modified the code snippet to add an example with Optimum pipelines.
It looks great!!!! Let's merge it :D
· Sign up or log in to comment
