From 11204f7fc953d2f64cd52d78cd5a8e35909b03da Mon Sep 17 00:00:00 2001 From: Eliott Coyac Date: Tue, 7 Mar 2023 09:19:22 +0000 Subject: [PATCH] Update README.md (#1) - Update README.md (71fdc45ba2a9d48167b9cfe5823d7c60c33b1d1f) Co-authored-by: Alara Dirik --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index c1e4639..a5bcb2b 100644 --- a/README.md +++ b/README.md @@ -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