|
|
from fastai.vision.all import * |
|
|
import gradio as gr |
|
|
|
|
|
|
|
|
model_inf=load_learner('export.pkl') |
|
|
|
|
|
categories=('Art Deco','Byzantine','Gothic','Neoclassical') |
|
|
|
|
|
def image_classifier(img): |
|
|
pred,pred_idx,probs = model_inf.predict(img) |
|
|
return dict(zip(categories,map(float,probs))) |
|
|
examples_arch=['ArtDeco.jpg','Byzantine.jpg','Gothic.jpg','Neoclassical.jpg'] |
|
|
intrf=intrf = gr.Interface( |
|
|
fn=image_classifier, |
|
|
inputs=gr.Image(width=192, height=192), |
|
|
outputs=gr.Label(num_top_classes=4), |
|
|
examples=examples_arch |
|
|
) |
|
|
intrf.launch(inline=False) |
