microsoft
/

beit-base-patch16-224

Image Classification

Model card Files Files and versions

nielsr HF Staff commited on Aug 14, 2021

Commit

c0f6a2a

·

1 Parent(s): fee5c23

Fix typo

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -38,7 +38,7 @@ 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')
-model = BEiTForImageClassification.from_pretrained('google/beit-base-patch16-224')
 inputs = feature_extractor(images=image, return_tensors="pt")
 outputs = model(**inputs)
 logits = outputs.logits

 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')
+model = BEiTForImageClassification.from_pretrained('microsoft/beit-base-patch16-224')
 inputs = feature_extractor(images=image, return_tensors="pt")
 outputs = model(**inputs)
 logits = outputs.logits