lasivia-tech/gamecon-tema

1import timm2from fastai.vision.all import *3import gradio as gr4 5learn_inf = load_learner('model.pkl')6 7categories = "2021 Mythos", "2023 ?", "2016 Mafie", "2019 Piráti", "2022 Duna", "2018 Western"8 9 10def classify_image(img):11    pred,idx,probs = learn_inf.predict(img)12    return dict(zip(categories, map(float,probs)))13 14image = gr.inputs.Image(shape=(192,192))15label = gr.outputs.Label()16examples = ["pirati.jpg", "western.jpg", "mythos.jpg", "duna.jpg", "mafie.jpg"]17title = "Veštič tématu GameConu"18description = "Chceš vědět jaké bude téma příštího GameConu? Zde můžeš zkusit nahrát tématický obrázek a ověřit tak, jak moc odpovídá různým ročníkům."19 20intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, title=title, description=description, examples=examples)21intf.launch(inline=False, debug=False)22 23