oschamp
/

vit-artworkclassifier

Image Classification

Generated from Trainer

Model card Files Files and versions

Metrics Training metrics Community

oschamp commited on Feb 22, 2023

Commit

26115da

·

1 Parent(s): 194a4fd

Update README.md

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

@@ -80,9 +80,9 @@ The following hyperparameters were used during training:
 ### Code to Run
 def vit_classify(image):
-    from transformers import ViTFeatureExtractor
-    from transformers import ViTForImageClassification
-    import torch
     vit = ViTForImageClassification.from_pretrained("oschamp/vit-artworkclassifier")
     vit.eval()

 ### Code to Run
 def vit_classify(image):
+    # from transformers import ViTFeatureExtractor
+    # from transformers import ViTForImageClassification
+    # import torch
     vit = ViTForImageClassification.from_pretrained("oschamp/vit-artworkclassifier")
     vit.eval()