Views
No views yet
1import requests
2from PIL import Image
3from transformers import BlipProcessor, BlipForConditionalGeneration
4
5# Load BLIP processor and model
6processor = BlipProcessor.from_pretrained("hannalaguilar/blip-all-clothing")
7model = BlipForConditionalGeneration.from_pretrained("hannalaguilar/blip-all-clothing")
8
9# Load model
10img_path = 'tshirt.jpg'
11image = Image.open(img_path).convert('RGB')
12
13
14# Process the image
15inputs = processor(images=image, return_tensors="pt")
16
17# Generate captions
18outputs = model.generate(**inputs)
19caption = processor.decode(outputs[0], skip_special_tokens=True)
20print("Generated Caption:", caption)
21