ankur310794 commonslash commited on
Commit
dc68f91
1 Parent(s): 0e334c7

Updates README.md sample running code to remove FutureWarning deprecation error for `ViTFeatureExtractor`. (#14)

Browse files

- Updates README.md sample running code to remove FutureWarning deprecation error for `ViTFeatureExtractor`. (496b5b1656cab6f6f963d1db47724f3cc56e6c7a)


Co-authored-by: E <[email protected]>

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")