Update README.md

README.md

@@ -1,13 +1,75 @@
 ---
-title: Painting Classifier Keras Final
-emoji: 📊
-colorFrom: blue
-colorTo: yellow
+title: WikiArt Painting Classifier
+emoji: 🎨
+colorFrom: purple
+colorTo: blue
 sdk: gradio
-sdk_version: 5.49.1
+sdk_version: 4.44.0
 app_file: app.py
 pinned: false
 license: mit
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# WikiArt Painting Classifier - Interactive Demo
+
+An interactive Gradio application for classifying paintings by artist, genre, and style.
+
+## Features
+
+- **Multi-Task Classification**: Predicts artist, genre, and style simultaneously
+- **129 Artists**: Including Monet, Van Gogh, Picasso, Da Vinci, Rembrandt, and more
+- **11 Genres**: Portrait, landscape, abstract painting, still life, cityscape, etc.
+- **27 Art Styles**: Impressionism, Cubism, Renaissance, Baroque, Expressionism, and more
+- **Real-Time Predictions**: Instant results with confidence scores
+
+## How to Use
+
+1. Upload a painting image using the interface
+2. Click "Classify Painting" or wait for automatic classification
+3. View predictions for:
+   - Top 10 likely artists with confidence scores
+   - Genre classification with probabilities
+   - Top 10 likely art styles with confidence scores
+
+## Model Details
+
+This app uses the model hosted at: [meetran/painting-classifier-keras-v1](https://huggingface.co/meetran/painting-classifier-keras-v1)
+
+### Architecture
+- **Base Model**: MobileNetV2 (pre-trained on ImageNet)
+- **Approach**: Multi-head classification with shared feature extraction
+- **Framework**: TensorFlow/Keras
+- **Input Size**: 224x224 RGB images
+
+### Training Details
+- **Dataset**: WikiArt dataset with 84,440 paintings
+- **Split**: 75% training, 25% validation
+- **Augmentation**: Random horizontal flip, rotation, zoom
+- **Training Stages**: Frozen backbone (2 epochs) + full fine-tuning (10 epochs)
+
+## Supported Classes
+
+### Artists (129 classes)
+Claude Monet, Vincent van Gogh, Pablo Picasso, Leonardo da Vinci, Rembrandt, Salvador Dali, Edgar Degas, Pierre-Auguste Renoir, Paul Cezanne, Henri Matisse, Michelangelo, Raphael, Edvard Munch, Gustav Klimt, Wassily Kandinsky, and 114 more.
+
+### Genres (11 classes)
+Abstract painting, Cityscape, Genre painting, Illustration, Landscape, Nude painting, Portrait, Religious painting, Sketch and study, Still life, Unknown Genre
+
+### Art Styles (27 classes)
+Impressionism, Post-Impressionism, Cubism, Abstract Expressionism, Early Renaissance, High Renaissance, Northern Renaissance, Mannerism Late Renaissance, Baroque, Rococo, Romanticism, Realism, Expressionism, Symbolism, Art Nouveau, Fauvism, Pointillism, Naive Art Primitivism, Synthetic Cubism, Analytical Cubism, Action painting, Color Field Painting, Contemporary Realism, New Realism, Minimalism, Pop Art, Ukiyo-e
+
+## Limitations
+
+- Best performance on Western classical and modern art from the training distribution
+- May have reduced accuracy on contemporary art, digital art, or non-Western styles
+- "Unknown Artist" and "Unknown Genre" classes may have lower precision
+- Requires clear, well-lit images for optimal results
+
+## License
+
+MIT License
+
+## Links
+
+- **Model Repository**: https://huggingface.co/meetran/painting-classifier-keras-v1
+- **Training Dataset**: [WikiArt on Hugging Face](https://huggingface.co/datasets/huggan/wikiart)