From b42d12b2efa706a0614ce2e4ba4c96445c3bbf20 Mon Sep 17 00:00:00 2001 From: Niels Rogge Date: Sat, 21 Aug 2021 10:40:28 +0000 Subject: [PATCH] Fix typo in code example --- README.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 023d311..35a8292 100644 --- a/README.md +++ b/README.md @@ -32,13 +32,13 @@ fine-tuned versions on a task that interests you. Here is how to use this model to classify an image of the COCO 2017 dataset into one of the 1,000 ImageNet classes: ```python -from transformers import BEiTFeatureExtractor, BEiTForImageClassification +from transformers import BeitFeatureExtractor, BeitForImageClassification from PIL import Image import requests url = 'http://images.cocodataset.org/val2017/000000039769.jpg' image = Image.open(requests.get(url, stream=True).raw) -feature_extractor = BEiTFeatureExtractor.from_pretrained('microsoft/beit-base-patch16-224-pt22k-ft22k') -model = BEiTForImageClassification.from_pretrained('microsoft/beit-base-patch16-224-pt22k-ft22k') +feature_extractor = BeitFeatureExtractor.from_pretrained('microsoft/beit-base-patch16-224-pt22k-ft22k') +model = BeitForImageClassification.from_pretrained('microsoft/beit-base-patch16-224-pt22k-ft22k') inputs = feature_extractor(images=image, return_tensors="pt") outputs = model(**inputs) logits = outputs.logits