Change to actual expect image class label

huggingface · May 26, 2023 · b188b39 · b188b39
1 parent d442667
commit b188b39
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/models/efficientformer/modeling_tf_efficientformer.py b/src/transformers/models/efficientformer/modeling_tf_efficientformer.py
@@ -55,7 +55,7 @@
 
 # Image classification docstring
 _IMAGE_CLASS_CHECKPOINT = "snap-research/efficientformer-l1-300"
-_IMAGE_CLASS_EXPECTED_OUTPUT = "Egyptian cat"
+_IMAGE_CLASS_EXPECTED_OUTPUT = "LABEL_281"
 
 
 TF_EFFICIENTFORMER_PRETRAINED_MODEL_ARCHIVE_LIST = [