Update README.md

README.md

@@ -2,8 +2,12 @@
 license: apache-2.0
 datasets:
 - Artificio/WikiArt
+library_name: transformers
 ---
 
+# **WikiArt-Style**
+
+> **WikiArt-Style** is a vision model fine-tuned from **google/siglip2-base-patch16-224** using the **SiglipForImageClassification** architecture. It classifies art images into one of 137 painting style categories.
 
 ```py
 from datasets import load_dataset
@@ -23,3 +27,251 @@ print(id2label)
 
     {'0': 'Abstract Art', '1': 'Abstract Expressionism', '2': 'Academicism', '3': 'Action painting', '4': 'American Realism', '5': 'Analytical Cubism', '6': 'Analytical\xa0Realism', '7': 'Art Brut', '8': 'Art Deco', '9': 'Art Informel', '10': 'Art Nouveau (Modern)', '11': 'Automatic Painting', '12': 'Baroque', '13': 'Biedermeier', '14': 'Byzantine', '15': 'Cartographic Art', '16': 'Classicism', '17': 'Cloisonnism', '18': 'Color Field Painting', '19': 'Conceptual Art', '20': 'Concretism', '21': 'Constructivism', '22': 'Contemporary Realism', '23': 'Costumbrismo', '24': 'Cubism', '25': 'Cubo-Expressionism', '26': 'Cubo-Futurism', '27': 'Dada', '28': 'Divisionism', '29': 'Early Renaissance', '30': 'Environmental (Land) Art', '31': 'Existential Art', '32': 'Expressionism', '33': 'Fantastic Realism', '34': 'Fauvism', '35': 'Feminist Art', '36': 'Figurative Expressionism', '37': 'Futurism', '38': 'Gongbi', '39': 'Gothic', '40': 'Hard Edge Painting', '41': 'High Renaissance', '42': 'Hyper-Realism', '43': 'Ilkhanid', '44': 'Impressionism', '45': 'Indian Space painting', '46': 'Ink and wash painting', '47': 'International Gothic', '48': 'Intimism', '49': 'Japonism', '50': 'Joseon Dynasty', '51': 'Kinetic Art', '52': 'Kitsch', '53': 'Lettrism', '54': 'Light and Space', '55': 'Luminism', '56': 'Lyrical Abstraction', '57': 'Magic Realism', '58': 'Mail Art', '59': 'Mannerism (Late Renaissance)', '60': 'Mechanistic Cubism', '61': 'Metaphysical art', '62': 'Minimalism', '63': 'Miserablism', '64': 'Modernismo', '65': 'Mosan art', '66': 'Muralism', '67': 'Nanga (Bunjinga)', '68': 'Nas-Taliq', '69': 'Native Art', '70': 'Naturalism', '71': 'Naïve Art (Primitivism)', '72': 'Neo-Byzantine', '73': 'Neo-Concretism', '74': 'Neo-Dada', '75': 'Neo-Expressionism', '76': 'Neo-Figurative Art', '77': 'Neo-Rococo', '78': 'Neo-Romanticism', '79': 'Neo-baroque', '80': 'Neoclassicism', '81': 'Neoplasticism', '82': 'New Casualism', '83': 'New European Painting', '84': 'New Realism', '85': 'Nihonga', '86': 'None', '87': 'Northern Renaissance', '88': 'Nouveau Réalisme', '89': 'Op Art', '90': 'Orientalism', '91': 'Orphism', '92': 'Ottoman Period', '93': 'Outsider art', '94': 'Perceptism ', '95': 'Photorealism', '96': 'Pointillism', '97': 'Pop Art', '98': 'Post-Impressionism', '99': 'Post-Minimalism', '100': 'Post-Painterly Abstraction', '101': 'Poster Art Realism', '102': 'Precisionism', '103': 'Primitivism', '104': 'Proto Renaissance', '105': 'Purism', '106': 'Rayonism', '107': 'Realism', '108': 'Regionalism', '109': 'Renaissance', '110': 'Rococo', '111': 'Romanesque', '112': 'Romanticism', '113': 'Safavid Period', '114': 'Shin-hanga', '115': 'Social Realism', '116': 'Socialist Realism', '117': 'Spatialism', '118': 'Spectralism', '119': 'Street art', '120': 'Suprematism', '121': 'Surrealism', '122': 'Symbolism', '123': 'Synchromism', '124': 'Synthetic Cubism', '125': 'Synthetism', '126': 'Sōsaku hanga', '127': 'Tachisme', '128': 'Tenebrism', '129': 'Timurid Period', '130': 'Tonalism', '131': 'Transautomatism', '132': 'Tubism', '133': 'Ukiyo-e', '134': 'Verism', '135': 'Yamato-e', '136': 'Zen'}
 
+
+The model predicts one of the following painting **style** categories:
+
+```
+0: Abstract Art  
+1: Abstract Expressionism  
+2: Academicism  
+3: Action painting  
+4: American Realism  
+5: Analytical Cubism  
+6: Analytical Realism  
+7: Art Brut  
+8: Art Deco  
+9: Art Informel  
+10: Art Nouveau (Modern)  
+11: Automatic Painting  
+12: Baroque  
+13: Biedermeier  
+14: Byzantine  
+15: Cartographic Art  
+16: Classicism  
+17: Cloisonnism  
+18: Color Field Painting  
+19: Conceptual Art  
+20: Concretism  
+21: Constructivism  
+22: Contemporary Realism  
+23: Costumbrismo  
+24: Cubism  
+25: Cubo-Expressionism  
+26: Cubo-Futurism  
+27: Dada  
+28: Divisionism  
+29: Early Renaissance  
+30: Environmental (Land) Art  
+31: Existential Art  
+32: Expressionism  
+33: Fantastic Realism  
+34: Fauvism  
+35: Feminist Art  
+36: Figurative Expressionism  
+37: Futurism  
+38: Gongbi  
+39: Gothic  
+40: Hard Edge Painting  
+41: High Renaissance  
+42: Hyper-Realism  
+43: Ilkhanid  
+44: Impressionism  
+45: Indian Space painting  
+46: Ink and wash painting  
+47: International Gothic  
+48: Intimism  
+49: Japonism  
+50: Joseon Dynasty  
+51: Kinetic Art  
+52: Kitsch  
+53: Lettrism  
+54: Light and Space  
+55: Luminism  
+56: Lyrical Abstraction  
+57: Magic Realism  
+58: Mail Art  
+59: Mannerism (Late Renaissance)  
+60: Mechanistic Cubism  
+61: Metaphysical art  
+62: Minimalism  
+63: Miserablism  
+64: Modernismo  
+65: Mosan art  
+66: Muralism  
+67: Nanga (Bunjinga)  
+68: Nas-Taliq  
+69: Native Art  
+70: Naturalism  
+71: Naïve Art (Primitivism)  
+72: Neo-Byzantine  
+73: Neo-Concretism  
+74: Neo-Dada  
+75: Neo-Expressionism  
+76: Neo-Figurative Art  
+77: Neo-Rococo  
+78: Neo-Romanticism  
+79: Neo-baroque  
+80: Neoclassicism  
+81: Neoplasticism  
+82: New Casualism  
+83: New European Painting  
+84: New Realism  
+85: Nihonga  
+86: None  
+87: Northern Renaissance  
+88: Nouveau Réalisme  
+89: Op Art  
+90: Orientalism  
+91: Orphism  
+92: Ottoman Period  
+93: Outsider art  
+94: Perceptism  
+95: Photorealism  
+96: Pointillism  
+97: Pop Art  
+98: Post-Impressionism  
+99: Post-Minimalism  
+100: Post-Painterly Abstraction  
+101: Poster Art Realism  
+102: Precisionism  
+103: Primitivism  
+104: Proto Renaissance  
+105: Purism  
+106: Rayonism  
+107: Realism  
+108: Regionalism  
+109: Renaissance  
+110: Rococo  
+111: Romanesque  
+112: Romanticism  
+113: Safavid Period  
+114: Shin-hanga  
+115: Social Realism  
+116: Socialist Realism  
+117: Spatialism  
+118: Spectralism  
+119: Street art  
+120: Suprematism  
+121: Surrealism  
+122: Symbolism  
+123: Synchromism  
+124: Synthetic Cubism  
+125: Synthetism  
+126: Sōsaku hanga  
+127: Tachisme  
+128: Tenebrism  
+129: Timurid Period  
+130: Tonalism  
+131: Transautomatism  
+132: Tubism  
+133: Ukiyo-e  
+134: Verism  
+135: Yamato-e  
+136: Zen  
+```
+
+---
+
+## **Run with Transformers 🤗**
+
+```bash
+pip install -q transformers torch pillow gradio
+```
+
+```python
+import gradio as gr
+from transformers import AutoImageProcessor, SiglipForImageClassification
+from PIL import Image
+import torch
+
+# Load model and processor
+model_name = "prithivMLmods/WikiArt-Style"  # Replace with your model path
+model = SiglipForImageClassification.from_pretrained(model_name)
+processor = AutoImageProcessor.from_pretrained(model_name)
+
+# Label mapping
+id2label = {
+    0: "Abstract Art", 1: "Abstract Expressionism", 2: "Academicism", 3: "Action painting",
+    4: "American Realism", 5: "Analytical Cubism", 6: "Analytical Realism", 7: "Art Brut",
+    8: "Art Deco", 9: "Art Informel", 10: "Art Nouveau (Modern)", 11: "Automatic Painting",
+    12: "Baroque", 13: "Biedermeier", 14: "Byzantine", 15: "Cartographic Art", 16: "Classicism",
+    17: "Cloisonnism", 18: "Color Field Painting", 19: "Conceptual Art", 20: "Concretism",
+    21: "Constructivism", 22: "Contemporary Realism", 23: "Costumbrismo", 24: "Cubism",
+    25: "Cubo-Expressionism", 26: "Cubo-Futurism", 27: "Dada", 28: "Divisionism",
+    29: "Early Renaissance", 30: "Environmental (Land) Art", 31: "Existential Art",
+    32: "Expressionism", 33: "Fantastic Realism", 34: "Fauvism", 35: "Feminist Art",
+    36: "Figurative Expressionism", 37: "Futurism", 38: "Gongbi", 39: "Gothic",
+    40: "Hard Edge Painting", 41: "High Renaissance", 42: "Hyper-Realism", 43: "Ilkhanid",
+    44: "Impressionism", 45: "Indian Space painting", 46: "Ink and wash painting",
+    47: "International Gothic", 48: "Intimism", 49: "Japonism", 50: "Joseon Dynasty",
+    51: "Kinetic Art", 52: "Kitsch", 53: "Lettrism", 54: "Light and Space", 55: "Luminism",
+    56: "Lyrical Abstraction", 57: "Magic Realism", 58: "Mail Art", 59: "Mannerism (Late Renaissance)",
+    60: "Mechanistic Cubism", 61: "Metaphysical art", 62: "Minimalism", 63: "Miserablism",
+    64: "Modernismo", 65: "Mosan art", 66: "Muralism", 67: "Nanga (Bunjinga)", 68: "Nas-Taliq",
+    69: "Native Art", 70: "Naturalism", 71: "Naïve Art (Primitivism)", 72: "Neo-Byzantine",
+    73: "Neo-Concretism", 74: "Neo-Dada", 75: "Neo-Expressionism", 76: "Neo-Figurative Art",
+    77: "Neo-Rococo", 78: "Neo-Romanticism", 79: "Neo-baroque", 80: "Neoclassicism",
+    81: "Neoplasticism", 82: "New Casualism", 83: "New European Painting", 84: "New Realism",
+    85: "Nihonga", 86: "None", 87: "Northern Renaissance", 88: "Nouveau Réalisme", 89: "Op Art",
+    90: "Orientalism", 91: "Orphism", 92: "Ottoman Period", 93: "Outsider art", 94: "Perceptism ",
+    95: "Photorealism", 96: "Pointillism", 97: "Pop Art", 98: "Post-Impressionism",
+    99: "Post-Minimalism", 100: "Post-Painterly Abstraction", 101: "Poster Art Realism",
+    102: "Precisionism", 103: "Primitivism", 104: "Proto Renaissance", 105: "Purism",
+    106: "Rayonism", 107: "Realism", 108: "Regionalism", 109: "Renaissance", 110: "Rococo",
+    111: "Romanesque", 112: "Romanticism", 113: "Safavid Period", 114: "Shin-hanga",
+    115: "Social Realism", 116: "Socialist Realism", 117: "Spatialism", 118: "Spectralism",
+    119: "Street art", 120: "Suprematism", 121: "Surrealism", 122: "Symbolism",
+    123: "Synchromism", 124: "Synthetic Cubism", 125: "Synthetism", 126: "Sōsaku hanga",
+    127: "Tachisme", 128: "Tenebrism", 129: "Timurid Period", 130: "Tonalism",
+    131: "Transautomatism", 132: "Tubism", 133: "Ukiyo-e", 134: "Verism", 135: "Yamato-e",
+    136: "Zen"
+}
+
+def classify_style(image):
+    """Predicts the artistic style of the input artwork."""
+    image = Image.fromarray(image).convert("RGB")
+    inputs = processor(images=image, return_tensors="pt")
+    with torch.no_grad():
+        outputs = model(**inputs)
+        probs = torch.nn.functional.softmax(outputs.logits, dim=1).squeeze().tolist()
+    return {id2label[i]: round(probs[i], 3) for i in range(len(probs))}
+
+# Gradio interface
+iface = gr.Interface(
+    fn=classify_style,
+    inputs=gr.Image(type="numpy"),
+    outputs=gr.Label(label="Style Prediction Scores"),
+    title="WikiArt-Style",
+    description="Upload an art image to predict its painting style category (e.g., Impressionism, Cubism, Baroque, etc.)."
+)
+
+if __name__ == "__main__":
+    iface.launch()
+```
+
+---
+
+# **Intended Use of WikiArt-Style**
+
+**1. Style Classification in Machine Learning Models**
+- Used as **labels** for training and evaluating models that classify artworks based on their artistic styles.
+- Ideal for deep learning applications involving **convolutional neural networks (CNNs)** or **transformer-based vision models**.
+
+**2. Style Transfer Applications**
+- Acts as a **style reference** for neural style transfer algorithms (e.g., applying "Baroque" or "Cubism" to photos).
+- Can guide users to select a target style from a curated list.
+
+**3. Dataset Annotation**
+- Used to **annotate** images in large datasets of paintings with consistent style names.
+- Ensures compatibility with datasets like WikiArt, Kaggle’s Painter by Numbers, or custom curation.
+
+**4. Educational and Exploratory Interfaces**
+- Powers interfaces or apps for **exploring art history**, with filterable and searchable styles.
+- Great for building **art recommender systems** or **virtual museums**.
+
+**5. Generative Art Prompting**
+- Assists in **text-to-image prompting** for generative models (e.g., Stable Diffusion, DALL·E) to specify desired styles.
+- Example: "Generate a portrait in the style of Neo-Expressionism."
+
+**6. Metadata Categorization in Art Databases**
+- Useful for tagging and organizing artworks by style in digital archives or NFT marketplaces.