commonslash commited on
Commit
496b5b1
1 Parent(s): 0e334c7

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

Browse files

Changes `ViTFeatureExtractor` to `ViTFeatureExtractor` in sample running code to resolve FutureWarning error: 'FutureWarning: The class ViTFeatureExtractor is deprecated and will be removed in version 5 of Transformers. Please use ViTImageProcessor instead.'

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