0-ma
/

vit-geometric-shapes-tiny

Image Classification

Model card Files Files and versions Community

0-ma commited on Sep 12

Commit

4eeeace

•

1 Parent(s): 87e2f24

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -28,7 +28,7 @@ pipeline_tag: image-classification
     from transformers import AutoImageProcessor, AutoModelForImageClassification
     import requests
     labels =  [
-        "Only text",
         "Circle",
         "Triangle",
         "Square",

     from transformers import AutoImageProcessor, AutoModelForImageClassification
     import requests
     labels =  [
+        "None",
         "Circle",
         "Triangle",
         "Square",