Model Details
Full Model IDnlpconnect/vit-gpt2-image-captioning
Pipeline / Taskimage-to-text
Librarytransformers
Downloads (all-time)215.9K
Likes927
Last Modified2/27/2023
Author / Orgnlpconnect
PrivateNo — public
⚡ Quick Usage (Python)
Using the 🤗 Transformers library. Install with pip install transformers
from transformers import pipeline
# Load the model
pipe = pipeline("image-to-text", model="nlpconnect/vit-gpt2-image-captioning")
# Run inference
result = pipe("Your input here")
print(result)🏷️ Tags
transformerspytorchvision-encoder-decoderimage-text-to-textimage-to-textimage-captioningdoi:10.57967/hf/0222license:apache-2.0endpoints_compatibleregion:us
More image-to-text Models
See all →🚀 Use This Model
Access model files, inference API, and full documentation on Hugging Face.
Open on Hugging Face →Browse Model Files ↗← Browse All Models🤖 Task: image-to-text
This model is designed for the image-to-text task. Explore more models for this use case.
All image-to-text Models →📊 Popularity
⬇ Downloads215.9K
❤️ Community Likes927
🛠️ Requirements
- →Install: pip install transformers
- →Python 3.8+ recommended for Transformers.
- →GPU (CUDA) speeds up inference significantly.
- →Use model.half() for fp16 on limited VRAM.