Hugging Face's logo

sunbv56
/
vit-gpt2-imagecaptioningfood

Image-to-Text
Transformers
PyTorch
TensorBoard
English
vision-encoder-decoder
image-text-to-text
Inference Endpoints
Model card Files Metrics Community

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

Downloads last month
9
Inference Examples
Image-to-Text
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.