Updates README.md sample running code to remove FutureWarning deprecation error for `ViTFeatureExtractor`.

#14
by commonslash - opened
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -28,12 +28,12 @@ This is an image captioning model trained by @ydshieh in [flax ](https://github.
28
 
29
  ```python
30
 
31
- from transformers import VisionEncoderDecoderModel, ViTFeatureExtractor, AutoTokenizer
32
  import torch
33
  from PIL import Image
34
 
35
  model = VisionEncoderDecoderModel.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
36
- feature_extractor = ViTFeatureExtractor.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
37
  tokenizer = AutoTokenizer.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
38
 
39
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
28
 
29
  ```python
30
 
31
+ from transformers import VisionEncoderDecoderModel, ViTImageProcessor, AutoTokenizer
32
  import torch
33
  from PIL import Image
34
 
35
  model = VisionEncoderDecoderModel.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
36
+ feature_extractor = ViTImageProcessor.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
37
  tokenizer = AutoTokenizer.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
38
 
39
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")