Hugging Face's logo

facebook
/
vit-msn-small

Image Feature Extraction
Transformers
PyTorch
vit_msn
vision
Model card Files
xet
Community
6

updated the AutoFeatureExtractor to AutoImageProcessor in model usage code

#6
by shivamgupta - opened
base: refs/heads/main
←
from: refs/pr/6
Discussion Files changed
+2
-2
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -37,7 +37,7 @@ import requests
37
  url = "http://images.cocodataset.org/val2017/000000039769.jpg"
38
  image = Image.open(requests.get(url, stream=True).raw)
39
 
40
- feature_extractor = AutoFeatureExtractor.from_pretrained("facebook/vit-msn-small")
41
  model = ViTMSNModel.from_pretrained("facebook/vit-msn-small")
42
  inputs = feature_extractor(images=image, return_tensors="pt")
43
  with torch.no_grad():
@@ -56,7 +56,7 @@ import requests
56
  url = "http://images.cocodataset.org/val2017/000000039769.jpg"
57
  image = Image.open(requests.get(url, stream=True).raw)
58
 
59
- feature_extractor = AutoFeatureExtractor.from_pretrained("facebook/vit-msn-small")
60
  model = ViTMSNForImageClassification.from_pretrained("facebook/vit-msn-small")
61
 
62
  ...
 
37
  url = "http://images.cocodataset.org/val2017/000000039769.jpg"
38
  image = Image.open(requests.get(url, stream=True).raw)
39
 
40
+ feature_extractor = AutoImageProcessor.from_pretrained("facebook/vit-msn-small")
41
  model = ViTMSNModel.from_pretrained("facebook/vit-msn-small")
42
  inputs = feature_extractor(images=image, return_tensors="pt")
43
  with torch.no_grad():
 
56
  url = "http://images.cocodataset.org/val2017/000000039769.jpg"
57
  image = Image.open(requests.get(url, stream=True).raw)
58
 
59
+ feature_extractor = AutoImageProcessor.from_pretrained("facebook/vit-msn-small")
60
  model = ViTMSNForImageClassification.from_pretrained("facebook/vit-msn-small")
61
 
62
  ...