jzhang533 commited on
Commit
258714c
·
1 Parent(s): e548940

using processor since FeatureExtractor is deprecating

Browse files
Files changed (1) hide show
  1. README.md +1 -4
README.md CHANGED
@@ -40,11 +40,8 @@ import requests
40
  url = 'http://images.cocodataset.org/val2017/000000039769.jpg'
41
  image = Image.open(requests.get(url, stream=True).raw)
42
  processor = AutoImageProcessor.from_pretrained("facebook/deit-tiny-patch16-224")
43
- model = AutoModelForImageClassification.from_pretrained("facebook/deit-tiny-patch16-224")
44
-
45
-
46
  model = ViTForImageClassification.from_pretrained('facebook/deit-tiny-patch16-224')
47
- inputs = feature_extractor(images=image, return_tensors="pt")
48
  outputs = model(**inputs)
49
  logits = outputs.logits
50
  # model predicts one of the 1000 ImageNet classes
 
40
  url = 'http://images.cocodataset.org/val2017/000000039769.jpg'
41
  image = Image.open(requests.get(url, stream=True).raw)
42
  processor = AutoImageProcessor.from_pretrained("facebook/deit-tiny-patch16-224")
 
 
 
43
  model = ViTForImageClassification.from_pretrained('facebook/deit-tiny-patch16-224')
44
+ inputs = processor(images=image, return_tensors="pt")
45
  outputs = model(**inputs)
46
  logits = outputs.logits
47
  # model predicts one of the 1000 ImageNet classes