updated the AutoFeatureExtractor to AutoImageProcessor in model usage code

changed AutoFeatureExtractor to AutoImageProcessor, as ViTFeatureExtractor is going to be deprecated in version 5 of transformers

![image.png](https://cdn-uploads.huggingface.co/production/uploads/65174ffc757417990dfe9425/ONLV78KiGxGlcxtmV5bIy.png)

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -37,7 +37,7 @@ import requests
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 image = Image.open(requests.get(url, stream=True).raw)
-feature_extractor = AutoFeatureExtractor.from_pretrained("facebook/vit-msn-small")
 model = ViTMSNModel.from_pretrained("facebook/vit-msn-small")
 inputs = feature_extractor(images=image, return_tensors="pt")
 with torch.no_grad():
@@ -56,7 +56,7 @@ import requests
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 image = Image.open(requests.get(url, stream=True).raw)
-feature_extractor = AutoFeatureExtractor.from_pretrained("facebook/vit-msn-small")
 model = ViTMSNForImageClassification.from_pretrained("facebook/vit-msn-small")
 ...

 url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 image = Image.open(requests.get(url, stream=True).raw)
+feature_extractor = AutoImageProcessor.from_pretrained("facebook/vit-msn-small")
 model = ViTMSNModel.from_pretrained("facebook/vit-msn-small")
 inputs = feature_extractor(images=image, return_tensors="pt")
 with torch.no_grad():
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 image = Image.open(requests.get(url, stream=True).raw)
+feature_extractor = AutoImageProcessor.from_pretrained("facebook/vit-msn-small")
 model = ViTMSNForImageClassification.from_pretrained("facebook/vit-msn-small")
 ...