nlpconnect/vit-gpt2-image-captioning
nlpconnect/vit-gpt2-image-captioning
1
0
Overview Code Discussions Pull Requests Wiki Activity

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

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


Co-authored-by: E <commonslash@users.noreply.huggingface.co>
This commit is contained in:
Ankur Singh 2023-02-27 15:00:09 +00:00 committed by system
parent 0e334c70dc
commit dc68f91c06
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@ -28,12 +28,12 @@ This is an image captioning model trained by @ydshieh in [flax ](https://github.
```python
from transformers import VisionEncoderDecoderModel, ViTFeatureExtractor, AutoTokenizer
from transformers import VisionEncoderDecoderModel, ViTImageProcessor, AutoTokenizer
import torch
from PIL import Image
model = VisionEncoderDecoderModel.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
feature_extractor = ViTFeatureExtractor.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
feature_extractor = ViTImageProcessor.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
tokenizer = AutoTokenizer.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")