google
/

vit-base-patch16-224-in21k

Image Feature Extraction

Model card Files Files and versions

nielsr HF Staff commited on Sep 13, 2021

Commit

6b6c585

·

1 Parent(s): 8140d7d

Fix typo

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -57,7 +57,7 @@ url = 'http://images.cocodataset.org/val2017/000000039769.jpg'
 image = Image.open(requests.get(url, stream=True).raw)
 feature_extractor = ViTFeatureExtractor.from_pretrained('google/vit-base-patch16-224-in21k')
-model = ViTModel.from_pretrained('google/vit-base-patch16-224-in21k')
 inputs = feature_extractor(images=image, return_tensors="np")
 outputs = model(**inputs)

 image = Image.open(requests.get(url, stream=True).raw)
 feature_extractor = ViTFeatureExtractor.from_pretrained('google/vit-base-patch16-224-in21k')
+model = FlaxViTModel.from_pretrained('google/vit-base-patch16-224-in21k')
 inputs = feature_extractor(images=image, return_tensors="np")
 outputs = model(**inputs)