michelecafagna26
/

blip-base-captioning-ft-hl-actions

image-text-to-text

image-captioning

Inference Endpoints

Model card Files Files and versions Community

michelecafagna26 commited on Jul 22, 2023

Commit

bbe4f57

·

1 Parent(s): b1ad2e2

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -38,8 +38,8 @@ import requests
 from PIL import Image
 from transformers import BlipProcessor, BlipForConditionalGeneration
-processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
-model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base").to("cuda")
 img_url = 'https://datasets-server.huggingface.co/assets/michelecafagna26/hl/--/default/train/0/image/image.jpg'
 raw_image = Image.open(requests.get(img_url, stream=True).raw).convert('RGB')

 from PIL import Image
 from transformers import BlipProcessor, BlipForConditionalGeneration
+processor = BlipProcessor.from_pretrained("michelecafagna26/blip-base-captioning-ft-hl-actions")
+model = BlipForConditionalGeneration.from_pretrained("michelecafagna26/blip-base-captioning-ft-hl-actions").to("cuda")
 img_url = 'https://datasets-server.huggingface.co/assets/michelecafagna26/hl/--/default/train/0/image/image.jpg'
 raw_image = Image.open(requests.get(img_url, stream=True).raw).convert('RGB')