Downloads · 30 days
4
1% of all-time downloads
vit-gpt2-imagecaptioningfood
sunbv56/vit-gpt2-imagecaptioningfood
vit-gpt2-imagecaptioningfood is a image-to-text model from sunbv56. Use it when you need a caption or text from an image. It is set up for transformers. The card lists the license as apache-2.0.
The model fined tuning with large data of API from bbcgoodfood.com
All-time downloads
425
Public
Repo size
9.7 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin2 GB · 51%
At a glance
- Task
- Image-to-Text
- Library
- transformers
- License
- apache-2.0
- Model type
- vision-encoder-decoder
- Access
- Public
- Created
- Jun 25, 2024
- Updated
- Jul 27, 2024
- SHA
- 38b84c62
Model card
From the Hugging Face model README
About model
The model fined tuning with large data of API from bbcgoodfood.com
How to Get Started with the Model
from transformers import AutoTokenizer, VisionEncoderDecoderModel, ViTFeatureExtractor
tokenizer = AutoTokenizer.from_pretrained("gpt2") # for text
feature_extractor = ViTFeatureExtractor.from_pretrained("google/vit-base-patch16-224") # for image
model = VisionEncoderDecoderModel.from_pretrained("sunbv56/vit-gpt2-imagecaptioningfood") # load model
Example code here
https://www.kaggle.com/code/thuntrngbnh/test-model-vit-gpt2-icf/notebook
- Task
- Image-to-Text
- Library
- transformers
- Type
- vision-encoder-decoder
- License
- apache-2.0
- Languages
- en
- Created
- Jun 25, 2024
- Updated
- Jul 27, 2024