kakaobrain
/
align-base
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update README.md (#1) 

- Update README.md (71fdc45ba2a9d48167b9cfe5823d7c60c33b1d1f)


Co-authored-by: Alara Dirik <adirik@users.noreply.huggingface.co>
This commit is contained in:
Eliott Coyac 2023-03-07 09:19:22 +00:00 committed by system
parent d332311443
commit 11204f7fc9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -33,7 +33,7 @@ model = AlignModel.from_pretrained("kakaobrain/align-base")
url = "http://images.cocodataset.org/val2017/000000039769.jpg"
image = Image.open(requests.get(url, stream=True).raw)
inputs = processor(text=["a photo of a cat", "a photo of a dog"], images=image, return_tensors="pt", padding=True)
inputs = processor(text=["a photo of a cat", "a photo of a dog"], images=image, return_tensors="pt")
outputs = model(**inputs)
logits_per_image = outputs.logits_per_image # this is the image-text similarity score