Update README.md

README.md

@@ -1,3 +1,62 @@
----
-license: apache-2.0
----
+---
+license: apache-2.0
+pipeline_tag: image-classification
+base_model:
+- apple/aimv2-large-patch14-448
+datasets:
+- Artificio/WikiArt
+- asahi417/wikiart-face
+- mattmdjaga/human_parsing_dataset
+- sugarquark/4o
+- zguo0525/ArtBench
+tags:
+- fashion
+---
+
+# Classification model
+
+## Inference
+
+```
+aimv2.forward(pixel_values, return_dict=True).last_hidden_state
+pooled = hidden_states.mean(dim=1)
+logits = classifier(pooled)
+```
+
+## Categories
+
+```
+[1] Art Nouveau
+[2] Baroque
+[3] Contemporary Realism
+[4] Expressionism
+[5] Early Renaissance
+[6] High Renaissance
+[7] Impressionism
+[8] Neoclassicism
+[9] Northern Renaissance
+[10] Photo
+[11] Pointillism
+[12] Post-Impressionism
+[13] Realism
+[14] Renaissance
+[15] Rococo
+[16] Romanticism
+[19] 4o
+[20] Cityscape
+[21] Figurative
+[22] Flower painting
+[23] Genre painting
+[24] Landscape
+[25] Nude painting
+[26] Portrait
+[27] Religious painting
+[28] Self-portrait
+[29] Still life
+[31] Hat
+[34] Top
+[37] Dress
+[39] Shoes
+[46] Bag
+[49] Outer
+```