from fastai.vision.all import load_learner
import gradio as gr

art_labels= [
    "Art Nouveau",
    "Constructivism Art",
    "Cubism Art",
    "Dadaism Art",
    "Fauvism Art",
    "Gothic Art",
    "Minimalism Art",
    "Pointillism Art",
    "Pop Art",
    "Prehistoric Art"
    ]

model = load_learner('models/artStyle_recognizer-v5.pk1')

def recognize_image(image):
    _, _, probs = model.predict(image)
    return dict(zip(art_labels, map(float, probs)))

image = gr.Image()
label = gr.Label()
example = [
    'Art Nouvaeu.jpg',
    'Constructivism art.jpg',
    'Fauvism art.jpg',
    'Gothic art.jpeg',
    'Minimalistic art.jpg' 
]

iface = gr.Interface(fn = recognize_image, inputs= image, outputs = label, examples = example)
iface.launch(inline=False)