apple
/

MobileCLIP-B

Model card Files Files and versions

pcuenq HF Staff commited on Mar 6, 2024

Commit

69955ff

·

verified ·

1 Parent(s): 90585cc

Update snippet

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -47,8 +47,8 @@ import torch
 from PIL import Image
 import mobileclip
-model, _, preprocess = mobileclip.create_model_and_transforms('mobileclip_s0', pretrained='/path/to/mobileclip_s0.pt')
-tokenizer = mobileclip.get_tokenizer('mobileclip_s0')
 image = preprocess(Image.open("docs/fig_accuracy_latency.png").convert('RGB')).unsqueeze(0)
 text = tokenizer(["a diagram", "a dog", "a cat"])

 from PIL import Image
 import mobileclip
+model, _, preprocess = mobileclip.create_model_and_transforms('mobileclip_b', pretrained='/path/to/mobileclip_b.pt')
+tokenizer = mobileclip.get_tokenizer('mobileclip_b')
 image = preprocess(Image.open("docs/fig_accuracy_latency.png").convert('RGB')).unsqueeze(0)
 text = tokenizer(["a diagram", "a dog", "a cat"])