Spaces:
Build error
Build error
File size: 1,816 Bytes
b237e3b fd9bf1e 9f88271 fd9bf1e 49a0962 fd9bf1e 49a0962 fd9bf1e 9f88271 fd9bf1e b237e3b 9f88271 71c6dd3 b237e3b 71c6dd3 9f88271 3117eb2 9f88271 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
from fastai.vision.all import *
import string
import gradio as gr
def getClassName(fileName):
fileName = fileName[:-4] # remove extension .jpg
names = fileName.split('_')
if(names[0] == "Albrecht"):
return "Albrecht Dürer"
artist = ''.join(name+" " for name in names if name[0] in string.ascii_letters)
return artist[:-1] #Remove last space
def predict(img):
img = PILImage.create(img)
pred, pred_idx, probs = learn.predict(img)
return {labels[i]: float(probs[i]) for i in range(len(labels))}
learn = load_learner('drawingstyle.pkl')
labels = learn.dls.vocab
title = "What's your drawing style?"
description = "An artistic style classifier trained on the <a href='https://www.kaggle.com/datasets/ikarus777/best-artworks-of-all-time'>best-artworks-of-all-time</a> dataset with fastai. Created as homework for lesson 2 of the course <a href='https://course.fast.ai/Lessons/lesson2.html'>Pratical Deep Learning for Coders</a>.<br>Do you feel like something is off? Check the <a href='https://www.kaggle.com/code/nerusskyhigh/artisticdrawingstyle' target='_blank'>Kaggle Notebook</a>! ;)"
article="<p style='text-align: center'><a href='https://github.com/NerusSkyhigh/practicaldeeplearning' target='_blank'>Check the repository!</a></br>You can find more test images in the original dataset. Only the first 100 images per artist were used to train the neural network. As the other images were discarded, you can use them to test the network.</p>"
examples = ['Vincent_van_Gogh_151.jpg', 'Leonardo_da_Vinci_121.jpg', 'Pablo_Picasso_164.jpg']
gr.Interface( fn=predict, inputs=gr.inputs.Image(shape=(512,512)), outputs=gr.outputs.Label(num_top_classes=3),
title=title, description=description, article=article, examples=examples).launch(share=True)
|