boudiafA commited on
Commit
4d7037d
·
verified ·
1 Parent(s): d948a7c

Document custom zero-shot labels

Browse files
Files changed (1) hide show
  1. README.md +9 -0
README.md CHANGED
@@ -121,6 +121,15 @@ for label, score in classifier.predict(image, top_k=5):
121
  print(label, score)
122
  ```
123
 
 
 
 
 
 
 
 
 
 
124
  ## Evaluate Zero-Shot Accuracy
125
 
126
  The dataset should be arranged like `torchvision.datasets.ImageFolder`:
 
121
  print(label, score)
122
  ```
123
 
124
+ Use custom candidate labels by passing them to `predict`:
125
+
126
+ ```python
127
+ custom_labels = ["cacao", "coffee", "mango", "olive", "sunflower"]
128
+
129
+ for label, score in classifier.predict(image, labels=custom_labels, top_k=5):
130
+ print(label, score)
131
+ ```
132
+
133
  ## Evaluate Zero-Shot Accuracy
134
 
135
  The dataset should be arranged like `torchvision.datasets.ImageFolder`: