From d3323114431b379e02938093db5b0bf9fa4424c2 Mon Sep 17 00:00:00 2001 From: Niels Rogge Date: Fri, 3 Mar 2023 12:51:40 +0000 Subject: [PATCH] Update README.md --- README.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 320908e..c1e4639 100644 --- a/README.md +++ b/README.md @@ -25,10 +25,10 @@ Pre-trained representations are becoming crucial for many NLP and perception tas from PIL import Image import requests -from transformers import ALIGNProcessor, ALIGNModel +from transformers import AlignProcessor, AlignModel -model = ALIGNModel.from_pretrained("kakaobrain/align-base") -processor = ALIGNProcessor.from_pretrained("kakaobrain/align-base") +processor = AlignProcessor.from_pretrained("kakaobrain/align-base") +model = AlignModel.from_pretrained("kakaobrain/align-base") url = "http://images.cocodataset.org/val2017/000000039769.jpg" image = Image.open(requests.get(url, stream=True).raw)