Commit
·
13aad52
1
Parent(s):
22a1024
Update README.md (#8)
Browse files- Update README.md (4a7a084d5dce710df8ebbe4ae421e1b41c95dda8)
Co-authored-by: Omkar Subhash Ghongade <itsog@users.noreply.huggingface.co>
README.md
CHANGED
|
@@ -29,6 +29,8 @@ This is an image captioning model trained by @ydshieh in [flax ](https://github.
|
|
| 29 |
```python
|
| 30 |
|
| 31 |
from transformers import VisionEncoderDecoderModel, ViTFeatureExtractor, AutoTokenizer
|
|
|
|
|
|
|
| 32 |
|
| 33 |
model = VisionEncoderDecoderModel.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
|
| 34 |
feature_extractor = ViTFeatureExtractor.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
|
|
|
|
| 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")
|