Downloads · 30 days
0
Image_Captioner
SowmyaAndela/Image_Captioner
Image_Captioner is a machine learning model from SowmyaAndela. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
!pip install transformers Pillow torch torchvision torchaudio from transformers import BlipProcessor, BlipForConditionalGeneration from PIL import Image processor = BlipProcessor.frompretrained("Salesforce/blip-image-…
Access
Public
Updated Sep 1, 2024
Repo size
—
Likes
0
Public
Repo makeup
Click a slice to open those files.
Other1.5 KB · 68%
At a glance
- Access
- Public
- Created
- Sep 1, 2024
- Updated
- Sep 1, 2024
- SHA
- 3092c22e
Model card
From the Hugging Face model README
Install the transformers library
!pip install transformers Pillow torch torchvision torchaudio from transformers import BlipProcessor, BlipForConditionalGeneration from PIL import Image
Initialize the processor and model from Hugging Face
processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base") model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
Load an image
image = Image.open("path_to_your_image.jpg")
Prepare the image
inputs = processor(image, return_tensors="pt")
Generate captions
outputs = model.generate(**inputs) caption = processor.decode(outputs[0],skip_special_tokens=True)
print("Generated Caption:", caption)
- Created
- Sep 1, 2024
- Updated
- Sep 1, 2024