Spaces:

labofsahil
/

image-classification

Sleeping

notsahil commited on Nov 15

Commit

e07b644

verified ·

1 Parent(s): e5ce8c6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import os
 from transformers import pipeline
-image_classifier = pipeline(task="zero-shot-image-classification", model="google/siglip-so400m-patch14-384")
 labels = ['street sweeping', 'litter pickup', 'pothole repair', 'parking enforcement']
 def image_mod(image):

 from transformers import pipeline
+image_classifier = pipeline(task="zero-shot-image-classification", model="google/siglip-large-patch16-384")
 labels = ['street sweeping', 'litter pickup', 'pothole repair', 'parking enforcement']
 def image_mod(image):