Update README.md

README.md

@@ -35,8 +35,8 @@ from PIL import Image
 import open_clip
 
 # Load model and preprocessing
-model, _, preprocess = open_clip.create_model_and_transforms('ViT-B-32', pretrained='datology-cls-vit-b-32')
-tokenizer = open_clip.get_tokenizer('ViT-B-32')
+model, _, preprocess = open_clip.create_model_and_transforms('hf-hub:DatologyAI/cls-opt-vit-b-32')
+tokenizer = open_clip.get_tokenizer('hf-hub:DatologyAI/cls-opt-vit-b-32')
 
 # Load image
 image = preprocess(Image.open("path/to/image.jpg")).unsqueeze(0)
@@ -71,7 +71,7 @@ from PIL import Image
 import open_clip
 
 # Load model
-model, _, preprocess = open_clip.create_model_and_transforms('ViT-B-32', pretrained='datology-cls-vit-b-32')
+model, _, preprocess = open_clip.create_model_and_transforms('hf-hub:DatologyAI/cls-opt-vit-b-32')
 model.eval()
 
 # Process image