Spaces:
Sleeping
Sleeping
File size: 693 Bytes
9a59b3e fec710d a7c81c7 b87141f 9a59b3e b87141f 043681c b87141f 9a59b3e b87141f 9a59b3e b87141f 11da19b 69070c7 a7c81c7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
from fastai.vision.all import *
import gradio as gr
description = 'this is a model that predicts whether an album is metal or pop based on its cover art'
def is_metal_album(x): return x[0].isupper()
learn = load_learner('model.pkl')
categories = ('Pop Album', 'Metal Album')
labels = learn.dls.vocab
def predict(img):
img = PILImage.create(img)
pred,pred_idx,probs = learn.predict(img)
return dict(zip(categories, map(float,probs)))
examples = ['metalkinggizz.png', 'popkinggizz.jpeg', 'TI.png','TI2.jpeg']
gr.Interface(fn=predict, inputs=gr.Image(height=512,width=512), outputs=gr.Label(num_top_classes=2), examples=examples, description=description).launch(share=True) |