Duplicate from Datasculptor/AIart_sources_of_inspiration

Co-authored-by: Dariusz Gross MLearning.ai <[email protected]>

.gitattributes

@@ -0,0 +1,34 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,13 @@
+---
+title: Identifying Painting Authors
+emoji: 🎨
+colorFrom: indigo
+colorTo: blue
+sdk: gradio
+sdk_version: 3.12.0
+app_file: app.py
+pinned: false
+duplicated_from: Datasculptor/AIart_sources_of_inspiration
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,29 @@
+import tensorflow as tf
+import pandas as pd
+import gradio as gr
+
+authors_df = pd.read_csv('authors.csv')
+labels = sorted(list(authors_df.name))
+
+model = tf.keras.models.load_model('efficientnetb0.h5')
+
+description = 'This is a DEMO that attempts to recognize the inspirations used by the AI art generator. After uploading a picture of an image, the application displays the predicted artist along with the probability of predicting the top three authors.The DEMO uses EfficientNetB0 convolutional neural network as a base model whose classifier was modified and trained the 8,000+ paintings from [Kaggle](https://www.kaggle.com/datasets/ikarus777/best-artworks-of-all-time) dataset. Model trained by osydorchuk89. Given the dataset limitations, the model only recognizes paintings of [50 artists](https://huggingface.co/spaces/osydorchuk/painting_authors/blob/main/authors.csv).'
+
+def predict_author(input):
+    if input is None:
+        return 'Please upload an image'
+    input = input.reshape((-1, 224, 224, 3))
+    prediction = model.predict(input).flatten()
+    confidences = {labels[i]: float(prediction[i]) for i in range(50)}
+    return confidences
+
+demo = gr.Interface(
+    title='the AI art generator sources of inspiration',
+    description=description,
+    fn=predict_author, 
+    inputs=gr.Image(shape=(224, 224)), 
+    outputs=gr.Label(num_top_classes=3),
+    examples=['test_pics/eva_miro.jpg', 'test_pics/eva_bosch.jpg',  'test_pics/eva_miro_2.jpg', 'test_pics/eva_rtology.jpg']
+    )
+             
+demo.launch()

authors.csv

@@ -0,0 +1,52 @@
+id,name
+0,Amedeo Modigliani
+1,Vasiliy Kandinskiy
+2,Diego Rivera
+3,Claude Monet
+4,Rene Magritte
+5,Salvador Dali
+6,Edouard Manet
+7,Andrei Rublev
+8,Vincent van Gogh
+9,Gustav Klimt
+10,Hieronymus Bosch
+11,Kazimir Malevich
+12,Mikhail Vrubel
+13,Pablo Picasso
+14,Peter Paul Rubens
+15,Pierre-Auguste Renoir
+16,Francisco Goya
+17,Frida Kahlo
+18,El Greco
+19,Albrecht Durer
+20,Alfred Sisley
+21,Pieter Bruegel
+22,Marc Chagall
+23,Giotto di Bondone
+24,Sandro Botticelli
+25,Caravaggio
+26,Leonardo da Vinci
+27,Diego Velazquez
+28,Henri Matisse
+29,Jan van Eyck
+30,Edgar Degas
+31,Rembrandt
+32,Titian
+33,Henri de Toulouse-Lautrec
+34,Gustave Courbet
+35,Camille Pissarro
+36,William Turner
+37,Edvard Munch
+38,Paul Cezanne
+39,Eugene Delacroix
+40,Henri Rousseau
+41,Georges Seurat
+42,Paul Klee
+43,Piet Mondrian
+44,Joan Miro
+45,Andy Warhol
+46,Paul Gauguin
+47,Raphael
+48,Michelangelo
+49,Jackson Pollock
+

efficientnetb0.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2b37633e2aa793bda5c6934ea002c10eed66c6973c31a80dc44ed7a669a1291a
+size 17362032

requirements.txt

@@ -0,0 +1,3 @@
+pandas
+tensorflow==2.9.0
+gradio