visheratin
/

nllb-clip-large-siglip

Zero-Shot Image Classification

Model card Files Files and versions Community

visheratin commited on Jan 19, 2024

Commit

1e9f46f

•

1 Parent(s): b7f24e7

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -37,9 +37,9 @@ from PIL import Image
 import requests
 import torch
-model, transform = create_model_from_pretrained("nllb-clip-base-siglip", "v1", device="cuda")
-tokenizer = get_tokenizer("nllb-clip-base-siglip")
 class_options = ["бабочка", "butterfly", "kat"]
 class_langs = ["rus_Cyrl", "eng_Latn", "afr_Latn"]

 import requests
 import torch
+model, transform = create_model_from_pretrained("nllb-clip-large-siglip", "v1", device="cuda")
+tokenizer = get_tokenizer("nllb-clip-large-siglip")
 class_options = ["бабочка", "butterfly", "kat"]
 class_langs = ["rus_Cyrl", "eng_Latn", "afr_Latn"]