liammcdevitt73/LoL-Support-Classifier

1from fastai.vision.all import *2import gradio as gr3 4learn = load_learner('lol_support_model.pkl')5 6categories = ("ALISTAR","ANIVIA","ASHE","BARD","BLITZCRANK","BRAUM","FIDDLESTICKS","HEIMERDINGER","IVERN","JANNA","KARMA","KAYLE","LEONA","LULU","LUX","MILIO","MORGANA","NAMI","NAUTILUS","NEEKO","ORIANNA","PYKE","RAKAN","RELL","RENATA GLASC","SENNA","SERAPHINE","SONA","SORAKA","TAHM KENCH","TALIYAH","TARIC","THRESH","YUUMI","ZILEAN","ZOE","ZYRA")7 8def classify_image(image):9    pred, idx, probs = learn.predict(image)10    return dict(zip(categories, map(float, probs)))11 12im = gr.inputs.Image(shape=(192, 192))13label = gr.outputs.Label()14examples = ['Alistar.jpg', 'Blitzcrank.jpg', 'Sona.jpg', 'Karma.jpg', 'Soraka.jpg', 'Zilean.jpg']15 16intf = gr.Interface(fn=classify_image, inputs=im, examples=examples, outputs=label)17intf.launch(inline=False)