kakaobrain
/

align-base

Zero-Shot Image Classification

Model card Files Files and versions

nielsr HF Staff commited on Mar 3, 2023

Commit

d332311

·

1 Parent(s): 5623896

Update README.md

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

@@ -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
-model = ALIGNModel.from_pretrained("kakaobrain/align-base")
-processor = ALIGNProcessor.from_pretrained("kakaobrain/align-base")
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 image = Image.open(requests.get(url, stream=True).raw)

 from PIL import Image
 import requests
+from transformers import AlignProcessor, AlignModel
+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)