lindritdev commited on
Commit
6e50416
·
verified ·
1 Parent(s): 8cffda4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from transformers import pipeline
3
 
4
  # Load models
5
  vit_classifier = pipeline("image-classification", model="lindritdev/vit-base-oxford-iiit-pets")
6
- clip_detector = pipeline(model="openai/clip-vit-large-patch14", task="zero-shot-image-classification")
7
 
8
  labels_oxford_pets = [
9
  'Siamese', 'Birman', 'shiba inu', 'staffordshire bull terrier', 'basset hound', 'Bombay', 'japanese chin',
 
3
 
4
  # Load models
5
  vit_classifier = pipeline("image-classification", model="lindritdev/vit-base-oxford-iiit-pets")
6
+ clip_detector = pipeline(model="google/siglip2-so400m-patch14-384", task="zero-shot-image-classification")
7
 
8
  labels_oxford_pets = [
9
  'Siamese', 'Birman', 'shiba inu', 'staffordshire bull terrier', 'basset hound', 'Bombay', 'japanese chin',