0-ma
/

resnet-geometric-shapes

Image Classification

Model card Files Files and versions Community

0-ma commited on Sep 12

Commit

bc09910

•

1 Parent(s): 0aaa2a8

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",